Class name difference between original and quantized version of Sigmoid

torch.compile FX (Functional Transformations)
1

Hello, i`m interested in fx quantization of pytorch model. In my project i need to register model by its class name, for example:
I define module class from GraphModule by calling node.target. For quantized version of ConvBnReLU output would be QuantizedConvBnReLU.

Despite quantized Sigmoid my method works for any quantized module because of “Quantized” prefix in module class name. But for quantized Sigmoid class name would be “Sigmoid”, not “QuantizedSigmoid”. Why and is it error in naming?

2

I checked realization of quantized activations and Sigmoid doesn’t contain method _get_name.