Struct ReduceNodeOp

Inheritance Relationships

Base Type

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