Struct ReduceNodeOp¶

Defined in File node_operators_unary.h

Inheritance Relationships¶

Base Type¶

public marian::UnaryNodeOp (Struct UnaryNodeOp)

Struct Documentation¶

struct ReduceNodeOp : public marian::UnaryNodeOp ¶

Public Functions

ReduceNodeOp(Expr a, int axis, ReduceNodeOpCode opCode)¶

NodeOps forwardOps()¶

NodeOps backwardOps()¶

Shape newShape(Expr a, int axis)¶

const std::string type()¶

const std::string color()¶

virtual size_t hash()¶

virtual bool equal(Expr node)¶

Public Members

int axis_¶

ReduceNodeOpCode opCode_¶

int reducedDim_¶

Friends

friend marian::ReduceNodeOp::SerializationHelpers