7c8dc76223
* Rename template config and move transformations to code snippets * Fixed documentation * Rename template config
41 lines
1.5 KiB
C++
41 lines
1.5 KiB
C++
// Copyright (C) 2018-2023 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#include "template_model_transformation.hpp"
|
|
|
|
#include "openvino/cc/pass/itt.hpp"
|
|
|
|
// ! [model_pass:template_transformation_cpp]
|
|
// template_function_transformation.cpp
|
|
|
|
bool ov::pass::MyModelTransformation::run_on_model(const std::shared_ptr<ov::Model>& f) {
|
|
RUN_ON_MODEL_SCOPE(MyModelTransformation);
|
|
// Example transformation code
|
|
NodeVector nodes;
|
|
|
|
// Traverse nGraph Function in topological order
|
|
for (auto& node : f->get_ordered_ops()) {
|
|
// Check that number of input and output ports are equal to 1
|
|
if (node->inputs().size() == 1 && node->outputs().size() == 1) {
|
|
// Check that input and output shape a fully defined (not dynamic) and number of consumers equal to 1
|
|
Input<Node> input = node->input(0);
|
|
Output<Node> output = node->output(0);
|
|
if (input.get_partial_shape().is_static() && output.get_partial_shape().is_static() &&
|
|
output.get_target_inputs().size() == 1) {
|
|
nodes.push_back(node);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Print types and names for collected nodes
|
|
for (auto& node : nodes) {
|
|
std::cout << "Type: " << node->get_type_info().name << std::endl
|
|
<< "Name: " << node->get_friendly_name() << std::endl;
|
|
}
|
|
|
|
// Return false because we didn't change nGraph Function
|
|
return false;
|
|
}
|
|
// ! [model_pass:template_transformation_cpp]
|