Class OprFootprint¶

Defined in File opr_footprint.h

Nested Relationships¶

Nested Types¶

Struct OprFootprint::Result

Class Documentation¶

class mgb::OprFootprint¶

footprint for operators

Public Functions

OprFootprint()¶

Result calc_footprint(cg::OperatorNodeBase *opr)¶: return footprint rst for associated opr.

uint64_t get_computation(cg::OperatorNodeBase *opr)¶: get computation of a given operator

struct Result¶

Public Members

size_t memory = 0¶: total input/output memory

uint64_t computation = 0¶: total number of arithmetic computations; zero value means no trait function available

TensorLayoutArray inp_layout¶

TensorShapeArray out_shape¶

mgb::Typeinfo *opr_type¶