25 #include <unordered_map>
28 #include "ngraph/pass/pass.hpp"
32 using visualize_tree_ops_map_t =
34 std::function<void(
const ngraph::Node&, std::ostream& ss)>>;
43 NGRAPH_RTTI_DECLARATION;
45 using node_modifiers_t =
46 std::function<void(
const Node& node, std::vector<std::string>& attributes)>;
48 node_modifiers_t nm =
nullptr,
49 bool dot_only =
false);
50 bool run_on_function(std::shared_ptr<ngraph::Function>)
override;
52 void set_ops_to_details(
const visualize_tree_ops_map_t& ops_map)
54 m_ops_to_details = ops_map;
58 void add_node_arguments(std::shared_ptr<Node> node,
59 std::unordered_map<Node*, HeightMap>& height_maps,
60 size_t& fake_node_ctr);
61 std::string add_attributes(std::shared_ptr<Node> node);
62 virtual std::string get_attributes(std::shared_ptr<Node> node);
63 virtual std::string get_node_name(std::shared_ptr<Node> node);
64 std::string get_constant_value(std::shared_ptr<Node> node,
size_t max_elements = 7);
68 std::stringstream m_ss;
70 std::set<std::shared_ptr<Node>> m_nodes_with_attributes;
71 visualize_tree_ops_map_t m_ops_to_details;
72 node_modifiers_t m_node_modifiers =
nullptr;
74 static const int max_jump_distance;
Definition: visualize_tree.hpp:41
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28