|
using | node_modifiers_t = std::function< void(const Node &node, std::vector< std::string > &attributes)> |
|
using | type_info_t = DiscreteTypeInfo |
|
|
| VisualizeTree (const std::string &file_name, node_modifiers_t nm=nullptr, bool dot_only=false) |
|
bool | run_on_function (std::shared_ptr< ngraph::Function >) override |
|
void | set_ops_to_details (const visualize_tree_ops_map_t &ops_map) |
|
bool | get_property (const PassPropertyMask &prop_mask) const |
| Check if this pass has all the pass properties.
|
|
void | set_name (const std::string &name) |
|
std::string | get_name () const |
|
void | set_callback (const param_callback &callback) |
| Set callback for particular transformation type. This method set global callback. For more details see PassConfig class documentation. More...
|
|
virtual void | set_pass_config (const std::shared_ptr< PassConfig > &pass_config) |
| Set PassConfig for particular transformation instance. More...
|
|
std::shared_ptr< PassConfig > | get_pass_config () |
| Allows to access PassConfig shared instance. More...
|
|
bool | m_transformation_callback (const std::shared_ptr< const Node > &node) |
| Applies callback for given node. By default callback returns false. This method remains here only for backward compatibility and will be removed after all transformations are moved to transformation_callback() method. More...
|
|
bool | transformation_callback (const std::shared_ptr< const Node > &node) |
| Applies callback for given node. By default callback returns false. More...
|
|
virtual const type_info_t & | get_type_info () const =0 |
|
|
| NGRAPH_RTTI_DECLARATION |
|
| NGRAPH_RTTI_DECLARATION |
|
|
void | add_node_arguments (std::shared_ptr< Node > node, std::unordered_map< Node *, HeightMap > &height_maps, size_t &fake_node_ctr) |
|
std::string | add_attributes (std::shared_ptr< Node > node) |
|
virtual std::string | get_attributes (std::shared_ptr< Node > node) |
|
virtual std::string | get_node_name (std::shared_ptr< Node > node) |
|
std::string | get_constant_value (std::shared_ptr< Node > node, size_t max_elements=7) |
|
void | render () const |
|
void | set_property (const PassPropertyMask &prop, bool value) |
|
|
std::stringstream | m_ss |
|
std::string | m_name |
|
std::set< std::shared_ptr< Node > > | m_nodes_with_attributes |
|
visualize_tree_ops_map_t | m_ops_to_details |
|
node_modifiers_t | m_node_modifiers = nullptr |
|
bool | m_dot_only |
|
|
static const int | max_jump_distance |
|
The documentation for this class was generated from the following file: