Struct TopKNodeOp

Inheritance Relationships

Base Types

Struct Documentation

struct TopKNodeOp : public marian::UnaryNodeOp, public marian::TupleNode

Public Functions

TopKNodeOp(Expr a, int k, int axis, bool descending = true)
Shape newShape(Expr a, int k, int axis)
void allocateTuple()
void allocate()
void freeTuple()
void free()
virtual Expr tupleView()
void forward()
void backward()
const std::string type()
virtual size_t hash()
virtual bool equal(Expr node)