Class ParamRedistributePass

Inheritance Relationships

Base Type

  • public Pass

Class Documentation

class mgb::gopt::ParamRedistributePass : public Pass

redistribute SharedDeviceTensor oprs

Redistribute parameters. For example, conv(x, w) * k may be replaced by conv(x, w*k).

Usually this pass is used before ParamFusePass.

Public Functions

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