Struct TopKNodeOp

• Defined in File node_operators_tuple.h

Inheritance Relationships

Base Types

• public marian::UnaryNodeOp (Struct UnaryNodeOp)

• public marian::TupleNode (Class TupleNode)

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)