Class ConvertBatchNormToElemwisePass

Inheritance Relationships

Base Type

  • public Pass

Class Documentation

class mgb::gopt::ConvertBatchNormToElemwisePass : public Pass

convert batch norm to elemwise For inference phase, cudnnbn = scale * (x - mean) / variance + bias In order to make the latter ParamDistributePass + ParamFusePass to do const folding better

Public Functions

const char *name() const override
void apply(OptState &opt) const override