class ngraph::OpSet¶
Overview¶
Run-time opset information. More…
#include <opset.hpp>
class OpSet
{
public:
// methods
std::set<NodeTypeInfo>::size_type size() const;
void insert(
const std::string& name,
const NodeTypeInfo& type_info,
FactoryRegistry<Node>::Factory factory
);
template <typename OP_TYPE>
void insert(const std::string& name);
template <typename OP_TYPE>
void insert();
const std::set<NodeTypeInfo>& get_types_info() const;
ngraph::Node* create(const std::string& name) const;
ngraph::Node* create_insensitive(const std::string& name) const;
bool contains_type(const NodeTypeInfo& type_info) const;
template <typename OP_TYPE>
bool contains_type() const;
bool contains_type(const std::string& name) const;
bool contains_type_insensitive(const std::string& name) const;
bool contains_op_type(const Node* node) const;
const std::set<NodeTypeInfo>& get_type_info_set() const;
ngraph::FactoryRegistry<ngraph::Node>& get_factory_registry();
};
Detailed Documentation¶
Run-time opset information.
Methods¶
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.
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.