Class IOSameShapeOperatorNode

Inheritance Relationships

Base Type

Class Documentation

class mgb::cg::mixin::IOSameShapeOperatorNode : public mgb::cg::mixin::OutshapePureByInshapeOpr

mixin class for operator whose all inputs and outputs are the same shape

Protected Functions

void get_output_var_shape(const TensorShapeArray &inp_shape, TensorShapeArray &out_shape) const final override

get output shapes from input shapes

Parameters
  • inp_shape: current input shape; each element matches an input var

  • out_shape: output shape; storage already allocated, and each element matches an output var

~IOSameShapeOperatorNode() = default