FXFloatFunctional

class torch.ao.nn.quantized.FXFloatFunctional(*args, **kwargs)

在 FX 图模式量化之前替换 FloatFunctional 模块的模块，因为 activation_post_process 将直接插入到顶层模块中

有效操作名称

• add

• cat

• mul

• add_relu

• add_scalar

• mul_scalar