¶

Class ConvertFormatPass

• Defined in File inference.h

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()