Public Member Functions | Static Protected Member Functions | Protected Attributes
ngraph::OpSet Class Reference

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::Nodecreate (const std::string &name) const
 Create the op named name using it's factory.
 
ngraph::Nodecreate_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::Nodem_factory_registry
 
std::set< NodeTypeInfom_op_types
 
std::map< std::string, NodeTypeInfom_name_type_info_map
 
std::map< std::string, NodeTypeInfom_case_insensitive_type_info_map
 

Detailed Description

Run-time opset information.


The documentation for this class was generated from the following file: