Class ConvertFormatPass

Inheritance Relationships

Base Type

  • public Pass

Class Documentation

class mgb::gopt::ConvertFormatPass : public Pass

convert tensor format to speed up inference on certain devices

Public Functions

const char *name() const override
ConvertFormatPass &set_var_replace_check_flag(VarReplaceCheckFlag flag)
void apply(OptState &opt) const override

Public Static Functions

std::unique_ptr<ConvertFormatPass> make_nhwcd4_converter()