Run-time opset information. More...
#include <opset.hpp>
Public Member Functions | |
std::set< NodeTypeInfo >::size_type | size () const |
void | insert (const std::string &name, const NodeTypeInfo &type_info, FactoryRegistry< Node >::Factory factory) |
Insert an op into the opset with a particular name and factory. | |
template<typename OP_TYPE > | |
void | insert (const std::string &name) |
Insert OP_TYPE into the opset with a special name and the default factory. | |
template<typename OP_TYPE > | |
void | insert () |
Insert OP_TYPE into the opset with the default name and factory. | |
const std::set< NodeTypeInfo > & | get_types_info () const |
ngraph::Node * | create (const std::string &name) const |
Create the op named name using it's factory. | |
ngraph::Node * | create_insensitive (const std::string &name) const |
Create the op named name using it's factory. | |
bool | contains_type (const NodeTypeInfo &type_info) const |
Return true if OP_TYPE is in the opset. | |
template<typename OP_TYPE > | |
bool | contains_type () const |
Return true if OP_TYPE is in the opset. | |
bool | contains_type (const std::string &name) const |
Return true if name is in the opset. | |
bool | contains_type_insensitive (const std::string &name) const |
Return true if name is in the opset. | |
bool | contains_op_type (const Node *node) const |
Return true if node's type is in the opset. | |
const std::set< NodeTypeInfo > & | get_type_info_set () const |
ngraph::FactoryRegistry< ngraph::Node > & | get_factory_registry () |
Static Protected Member Functions | |
static std::string | to_upper_name (const std::string &name) |
Protected Attributes | |
ngraph::FactoryRegistry< ngraph::Node > | m_factory_registry |
std::set< NodeTypeInfo > | m_op_types |
std::map< std::string, NodeTypeInfo > | m_name_type_info_map |
std::map< std::string, NodeTypeInfo > | m_case_insensitive_type_info_map |
Run-time opset information.