class ov::pass::DepthToSpaceFusion¶
Overview¶
DepthToSpaceFusion transformation detects Reshape-Transpose-Reshape pattern and tries to fuse it into a single DepthToSpace layer. More…
#include <depth_to_space_fusion.hpp>
class DepthToSpaceFusion: public ov::pass::MatcherPass
{
public:
// methods
"DepthToSpaceFusion""0" OPENVINO_RTTI(, );
};
Inherited Members¶
public:
// typedefs
typedef DiscreteTypeInfo type_info_t;
// methods
boolconst PassPropertyMask& get_property() const;
voidconst std::string& set_name();
std::string get_name() const;
voidconst param_callback& set_callback();
virtual voidconst std::shared_ptr<PassConfig>& set_pass_config();
std::shared_ptr<PassConfig> get_pass_config();
boolconst std::shared_ptr<const Node>& transformation_callback();
virtual const type_info_t& get_type_info() const = 0;
"ov::pass::MatcherPass" OPENVINO_RTTI();
MatcherPass&const MatcherPass& operator = ();
boolstd::shared_ptr<ov::Node> apply();
template <, >
std::shared_ptr<T>Args&&... register_new_node();
template <>
std::shared_ptr<T>const std::shared_ptr<T>& register_new_node();
std::shared_ptr<ov::Node>const std::shared_ptr<ov::Node>& register_new_node_();
const std::vector<std::shared_ptr<ov::Node>>& get_new_nodes();
void clear_new_nodes();
std::shared_ptr<pattern::Matcher> get_matcher();
Detailed Documentation¶
DepthToSpaceFusion transformation detects Reshape-Transpose-Reshape pattern and tries to fuse it into a single DepthToSpace layer.
DepthToSpaceFusion transformation is optional and disabled by default. The transformation can be enabled with callback using setCallback method. See the example below.
Callback example:
// This callback enables DepthToSpaceFusion transformation
auto callback = [](const std::shared_ptr<const ov::Node> & node) -> bool {
return std::dynamic_pointer_cast<const ov::opset3::DepthToSpace>(node) != nullptr;
};
auto p = ov::pass::DepthToSpaceFusion();
p.setCallback(callback);
p.run_on_function(f);