Struct GatherNodeOp¶

Defined in File node_operators_binary.h

Inheritance Relationships¶

Base Type¶

public marian::NaryNodeOp (Struct NaryNodeOp)

Struct Documentation¶

struct GatherNodeOp : public marian::NaryNodeOp ¶

Public Functions

GatherNodeOp(Expr a, int axis, Expr indices)¶

NodeOps forwardOps()¶

NodeOps backwardOps()¶

Shape newShape(Expr a, int axis, Expr indices)¶

const std::string type()¶

const std::string color()¶

virtual size_t hash()¶

virtual bool equal(Expr node)¶