MatcherPass is a basic block for pattern based transformations. It describes pattern and action that is applied if pattern is matched.
More...
|
enum | LOG_TYPE { _LOG_TYPE_ERROR,
_LOG_TYPE_WARNING,
_LOG_TYPE_INFO,
_LOG_TYPE_DEBUG
} |
|
using | HostTensor = runtime::HostTensor |
|
using | HostTensorPtr = std::shared_ptr< HostTensor > |
|
using | HostTensorVector = std::vector< HostTensorPtr > |
|
using | ResultVector = std::vector< std::shared_ptr< op::v0::Result > > |
|
using | NodeMap = std::unordered_map< ngraph::Node *, std::shared_ptr< ngraph::Node > > |
| Alias useful for cloning.
|
|
using | NodeTypeInfo = Node::type_info_t |
|
using | RawNodeOutputMap = std::map< RawNodeOutput, Output< Node > > |
|
using | ParameterVector = std::vector< std::shared_ptr< op::Parameter > > |
|
using | ResultVector = std::vector< std::shared_ptr< op::Result > > |
|
using | NodeVector = std::vector< std::shared_ptr< Node > > |
|
using | OutputVector = std::vector< Output< Node > > |
|
using | matcher_pass_callback = std::function< bool(ngraph::pattern::Matcher &m)> |
|
using | graph_rewrite_callback = std::function< bool(ngraph::pattern::Matcher &m)> |
|
using | recurrent_graph_rewrite_callback = std::function< bool(ngraph::pattern::RecurrentMatcher &m)> |
|
using | handler_callback = std::function< bool(const std::shared_ptr< Node > &node)> |
|
using | node_modifiers_t = std::function< void(const Node &node, std::vector< std::string > &attributes)> |
|
using | Rank = Dimension |
| Alias for Dimension, used when the value represents the number of axes in a shape, rather than the size of one dimension in a shape. More...
|
|
using | NodeExecutorTy = std::function< void(const std::vector< void * > &inputs, std::vector< void * > &outputs)> |
|
using | BuildNodeExecutor = std::function< NodeExecutorTy(const ngraph::Node *)> |
|
using | BuildNodeExecutorMap = std::unordered_map< std::type_index, BuildNodeExecutor > |
|
using | VariantTypeInfo = DiscreteTypeInfo |
|
|
| Duration (const std::string &name, const std::string &category, const std::string &args="") |
|
void | stop () |
| stop the timer without writing the data to the log file. To write the data call the write method Calls to stop() are optional
|
|
void | write () |
| write the log data to the log file for this event This funtion has an implicit stop() if stop() has not been previously called
|
|
| Duration (const Duration &)=delete |
|
Duration & | operator= (Duration const &)=delete |
|
| ConstantFolding (const ngraph::BuildNodeExecutorMap &cfmap=ngraph::BuildNodeExecutorMap()) |
|
| ConvertFP32ToFP16 () |
|
| MatcherPass ()=default |
|
| MatcherPass (const MatcherPass &)=delete |
|
MatcherPass & | operator= (const MatcherPass &)=delete |
|
| MatcherPass (const std::string &name, const std::shared_ptr< pattern::Matcher > &m, const handler_callback &handler, const PassPropertyMask &property=PassProperty::CHANGE_DYNAMIC_STATE) |
|
bool | apply (std::shared_ptr< ngraph::Node > node) |
|
template<typename T , class... Args> |
std::shared_ptr< T > | register_new_node (Args &&... args) |
|
const std::vector< std::shared_ptr< ngraph::Node > > & | get_new_nodes () |
|
void | clear_new_nodes () |
|
std::shared_ptr< pattern::Matcher > | get_matcher () |
|
| GraphRewrite ()=default |
|
| GraphRewrite (const std::shared_ptr< MatcherPass > &pass) |
|
template<typename T , class... Args> |
std::shared_ptr< T > | add_matcher (Args &&... args) |
|
void | add_matcher (const std::shared_ptr< pattern::Matcher > &m, const ngraph::graph_rewrite_callback &callback, const PassPropertyMask &property) |
|
void | add_matcher (const std::shared_ptr< pattern::Matcher > &m, const ngraph::graph_rewrite_callback &callback) |
|
bool | run_on_function (std::shared_ptr< ngraph::Function > f) override |
|
| RecurrentGraphRewrite (size_t num_iters=10) |
|
void | add_matcher (const std::shared_ptr< pattern::RecurrentMatcher > &m, const ngraph::recurrent_graph_rewrite_callback &callback, const PassPropertyMask &property) |
|
void | add_matcher (const std::shared_ptr< pattern::RecurrentMatcher > &m, const ngraph::recurrent_graph_rewrite_callback &callback) |
|
virtual bool | run_on_function (std::shared_ptr< ngraph::Function > f) |
|
| Manager () |
|
template<typename T , class... Args> |
std::shared_ptr< T > | register_pass (Args &&... args) |
|
void | run_passes (std::shared_ptr< Function >) |
|
void | set_pass_visualization (bool new_state) |
|
void | set_per_pass_validation (bool new_state) |
| Set flag to enable/disable running Validate pass after executing each registered pass. More...
|
|
void | set_callback (param_callback callback) |
| Callback is a lambda function that can be used by registered transformations. The main purpose of this callback is to provide a way for plugins to disable/enable transformations. In some cases plugins may want not to execute some transformations. For example plugin can disable unpleasant decompositions because of performance reasons. Callback example: auto callback = [](const std::shared_ptr<const ngraph::Node> & node) -> bool { return std::dynamic_pointer_cast<const ngraph::opset3::DepthToSpace>(node) != nullptr; }; This callback returns true in case of DepthToSpace operation. So when execution DepthToSpace decomposition pass will check is this decomposition needed or plugin can execute this operation directly. And of course on transformation side we need to have a response for this callback. if (m_transformation_callback(batch_to_space)) { return false; }. More...
|
|
| PassConfig () |
|
const std::map< std::string, bool > & | get_enables () const |
|
void | set_pass_enable (const std::string &name, bool enable) |
|
bool | get_pass_enable (const std::string &name) const |
|
const std::map< std::string, bool > & | get_pass_attributes () const |
|
void | set_pass_attribute (const std::string &name, bool enable) |
|
bool | get_pass_attribute (const std::string &name) const |
|
| 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) |
|
| AlignedBuffer (size_t byte_size, size_t alignment=64) |
|
| AlignedBuffer () |
|
| AlignedBuffer (AlignedBuffer &&other) |
|
AlignedBuffer & | operator= (AlignedBuffer &&other) |
|
size_t | size () const |
|
void * | get_ptr (size_t offset) const |
|
void * | get_ptr () |
|
const void * | get_ptr () const |
|
template<typename T > |
T * | get_ptr () |
|
template<typename T > |
const T * | get_ptr () const |
|
template<typename T > |
| operator T* () |
|
| HostTensor (const element::Type &element_type, const Shape &shape, void *memory_pointer, const std::string &name="") |
|
| HostTensor (const element::Type &element_type, const Shape &shape, const std::string &name="") |
|
| HostTensor (const element::Type &element_type, const PartialShape &partial_shape, const std::string &name="") |
|
| HostTensor (const std::string &name="") |
|
| HostTensor (const Output< Node > &) |
|
| HostTensor (const std::shared_ptr< op::v0::Constant > &constant) |
|
void | initialize (const std::shared_ptr< op::v0::Constant > &constant) |
|
void * | get_data_ptr () |
|
const void * | get_data_ptr () const |
|
template<typename T > |
T * | get_data_ptr () |
|
template<typename T > |
const T * | get_data_ptr () const |
|
template<element::Type_t ET> |
element_type_traits< ET >::value_type * | get_data_ptr () |
|
template<element::Type_t ET> |
const element_type_traits< ET >::value_type * | get_data_ptr () const |
|
void | write (const void *p, size_t n) override |
| Write bytes directly into the tensor. More...
|
|
void | read (void *p, size_t n) const override |
| Read bytes directly from the tensor. More...
|
|
bool | get_is_allocated () const |
|
void | set_element_type (const element::Type &element_type) |
| Set the element type. Must be compatible with the current element type. More...
|
|
void | set_shape (const Shape &shape) |
| Set the actual shape of the tensor compatibly with the partial shape. More...
|
|
void | set_unary (const HostTensorPtr &arg) |
| Set the shape of a node from an input. More...
|
|
void | set_broadcast (const op::AutoBroadcastSpec &autob, const HostTensorPtr &arg0, const HostTensorPtr &arg1) |
| Set the shape of the tensor using broadcast rules. More...
|
|
void | set_broadcast (const op::AutoBroadcastSpec &autob, const HostTensorPtr &arg0, const HostTensorPtr &arg1, const element::Type &element_type) |
| Set the shape of the tensor using broadcast rules. More...
|
|