Class TensorReformatPass¶

Defined in File inference.h

Inheritance Relationships¶

Base Type¶

public Pass

Derived Types¶

public mgb::gopt::EnableCHWN4Pass (Class EnableCHWN4Pass)
public mgb::gopt::EnableNCHW4Pass (Class EnableNCHW4Pass)
public mgb::gopt::EnableNchwxxPass (Class EnableNchwxxPass)
public mgb::gopt::EnableTensorCorePass (Class EnableTensorCorePass)

Class Documentation¶

class mgb::gopt::TensorReformatPass : public Pass¶

tensor format converter to accelerate inference speed on Nvidia platform

Subclassed by mgb::gopt::EnableCHWN4Pass, mgb::gopt::EnableNCHW4Pass, mgb::gopt::EnableNchwxxPass, mgb::gopt::EnableTensorCorePass

Public Functions

TensorReformatPass &set_var_replace_check_flag(VarReplaceCheckFlag flag)¶

void apply(OptState &opt) const override¶

Protected Attributes

ThinHashMap<Typeinfo*, thin_function<OperatorNodeBase*(OperatorNodeBase*, const VarNodeArray&)>> m_opr_replace_func¶

VarReplaceCheckFlag m_var_replace_check_flag = VarReplaceCheckFlag::CHECK_ALL¶