Struct NaryNodeOp

• Defined in File node.h

Inheritance Relationships

Base Type

• public marian::Node (Class Node)

Derived Types

• public marian::AffineNodeOp (Class AffineNodeOp)

• public marian::AffineWithReluNodeOp (Class AffineWithReluNodeOp)

• public marian::ColsNodeOp (Struct ColsNodeOp)

• public marian::ConcatenateNodeOp (Struct ConcatenateNodeOp)

• public marian::cpu::variant::FbgemmPacked16AffineNodeOp (Class FbgemmPacked16AffineNodeOp)

• public marian::cpu::variant::FbgemmPacked8AffineNodeOp (Class FbgemmPacked8AffineNodeOp)

• public marian::CrossEntropyNodeOp (Class CrossEntropyNodeOp)

• public marian::CSRDotNodeOp (Class CSRDotNodeOp)

• public marian::DotBatchedLegacyNodeOp (Class DotBatchedLegacyNodeOp)

• public marian::DotBatchedNodeOp (Class DotBatchedNodeOp)

• public marian::DotNodeOp (Class DotNodeOp)

• public marian::ElementBinaryNodeOp (Struct ElementBinaryNodeOp)

• public marian::GatherNodeOp (Struct GatherNodeOp)

• public marian::HighwayNodeOp (Struct HighwayNodeOp)

• public marian::LambdaNodeOp (Class LambdaNodeOp)

• public marian::LayerNormalizationOp (Struct LayerNormalizationOp)

• public marian::RMSNormalizationOp (Struct RMSNormalizationOp)

• public marian::rnn::AttentionNodeOp (Struct AttentionNodeOp)

• public marian::rnn::GRUFastNodeOp (Struct GRUFastNodeOp)

• public marian::rnn::LSTMCellNodeOp (Struct LSTMCellNodeOp)

• public marian::rnn::LSTMOutputNodeOp (Struct LSTMOutputNodeOp)

• public marian::RowsNodeOp (Struct RowsNodeOp)

• public marian::ScalarProductNodeOp (Struct ScalarProductNodeOp)

• public marian::ScatterNodeOp (Struct ScatterNodeOp)

• public marian::TanhNodeOp (Struct TanhNodeOp)

• public marian::UnaryNodeOp (Struct UnaryNodeOp)

Struct Documentation

struct NaryNodeOp : public marian::Node

Subclassed by marian::AffineNodeOp, marian::AffineWithReluNodeOp, marian::ColsNodeOp, marian::ConcatenateNodeOp, marian::cpu::variant::FbgemmPacked16AffineNodeOp, marian::cpu::variant::FbgemmPacked8AffineNodeOp, marian::CrossEntropyNodeOp, marian::CSRDotNodeOp, marian::DotBatchedLegacyNodeOp, marian::DotBatchedNodeOp, marian::DotNodeOp, marian::ElementBinaryNodeOp, marian::GatherNodeOp, marian::HighwayNodeOp, marian::LambdaNodeOp, marian::LayerNormalizationOp, marian::RMSNormalizationOp, marian::rnn::AttentionNodeOp, marian::rnn::GRUFastNodeOp, marian::rnn::LSTMCellNodeOp, marian::rnn::LSTMOutputNodeOp, marian::RowsNodeOp, marian::ScalarProductNodeOp, marian::ScatterNodeOp, marian::TanhNodeOp, marian::UnaryNodeOp

Public Functions

NaryNodeOp(const std::vector<Expr> &nodes)

NaryNodeOp(const std::vector<Expr> &nodes, Shape shape)

NaryNodeOp(const std::vector<Expr> &nodes, Shape shape, Type value_type)

virtual ~NaryNodeOp()

std::vector<Expr> &children()

virtual size_t hash()

virtual bool equal(Expr node)

Public Members

size_t hash_ = {0}

Public Static Functions

static Type commonType(const std::vector<Expr> &nodes)