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.