class ov::pass::VisualizeTree¶
VisualizeTree pass allows to serialize ov::Model to xDot format.
#include <visualize_tree.hpp>
class VisualizeTree: public ov::pass::ModelPass
{
public:
// typedefs
typedef std::function<void(const Node&node, std::vector<std::string>&attributes)> node_modifiers_t;
// construction
VisualizeTree(
const std::string& file_name,
node_modifiers_t nm = nullptr,
bool dot_only = false
);
// methods
OPENVINO_RTTI("ov::pass::VisualizeTree");
virtual bool run_on_model(const std::shared_ptr<ov::Model>&);
void set_ops_to_details(const visualize_tree_ops_map_t& ops_map);
};Inherited Members¶
public:
// typedefs
typedef DiscreteTypeInfo type_info_t;
// methods
bool get_property(const PassPropertyMask& prop_mask) const;
void set_name(const std::string& name);
std::string get_name() const;
void set_callback(const param_callback& callback);
virtual void set_pass_config(const std::shared_ptr<PassConfig>& pass_config);
std::shared_ptr<PassConfig> get_pass_config();
bool m_transformation_callback(const std::shared_ptr<const Node>& node);
bool transformation_callback(const std::shared_ptr<const Node>& node);
virtual const type_info_t& get_type_info() const = 0;
OPENVINO_RTTI("ov::pass::ModelPass");
virtual bool run_on_function(std::shared_ptr<ov::Model> m);
virtual bool run_on_model(const std::shared_ptr<ov::Model>& m);