Struct AttentionNodeOp

• Defined in File attention.cpp

Inheritance Relationships

Base Type

• public marian::NaryNodeOp (Struct NaryNodeOp)

Struct Documentation

struct AttentionNodeOp : public marian::NaryNodeOp

Public Functions

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

Shape newShape(const std::vector<Expr> &nodes)

NodeOps forwardOps()

NodeOps backwardOps()

virtual void runBackward(const NodeOps &ops)

const std::string type()

const std::string color()