| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
| H A D | activation.py | 85 def from_reference(cls, mod, scale, zero_point): member in Hardswish
117 def from_reference(cls, mod, scale, zero_point): member in ELU
158 def from_reference(cls, mod, scale, zero_point): member in LeakyReLU
224 def from_reference(cls, mod, scale, zero_point): member in Softmax
329 def from_reference(cls, mod, scale, zero_point): member in PReLU
|
| H A D | normalization.py | 76 def from_reference(cls, mod, scale, zero_point): member in LayerNorm
203 def from_reference(cls, mod, scale, zero_point): member in InstanceNorm1d
270 def from_reference(cls, mod, scale, zero_point): member in InstanceNorm2d
337 def from_reference(cls, mod, scale, zero_point): member in InstanceNorm3d
|
| H A D | conv.py | 326 def from_reference(cls, ref_qconv, output_scale, output_zero_point): member in _ConvNd
844 def from_reference(cls, ref_qconvt, output_scale, output_zero_point): member in _ConvTransposeNd
992 def from_reference(cls, ref_qconvt, output_scale, output_zero_point): member in ConvTranspose1d
1115 def from_reference(cls, ref_qconvt, output_scale, output_zero_point): member in ConvTranspose2d
1240 def from_reference(cls, ref_qconvt, output_scale, output_zero_point): member in ConvTranspose3d
|
| H A D | embedding_ops.py | 247 def from_reference(cls, ref_embedding): member in Embedding
392 def from_reference(cls, ref_embedding_bag): member in EmbeddingBag
|
| H A D | dropout.py | 29 def from_reference(cls, mod, scale, zero_point): member in Dropout
|
| H A D | utils.py | 20 def from_reference(cls, ref_module, output_scale, output_zero_point): member in WeightedQuantizedModule
|
| H A D | batchnorm.py | 34 def from_reference(cls, bn, output_scale, output_zero_point): member in _BatchNorm
|
| H A D | linear.py | 343 def from_reference(cls, ref_qlinear, output_scale, output_zero_point): member in Linear
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/quantized/modules/ |
| H A D | linear_relu.py | 51 def from_reference(cls, ref_linear_relu, output_scale, output_zero_point): member in LinearReLU
117 def from_reference(cls, ref_mod, output_scale, output_zero_point): member in LinearLeakyReLU
180 def from_reference(cls, ref_mod, output_scale, output_zero_point): member in LinearTanh
|
| H A D | conv_relu.py | 95 def from_reference(cls, ref_qconv, output_scale, output_zero_point): member in ConvReLU1d
177 def from_reference(cls, ref_qconv, output_scale, output_zero_point): member in ConvReLU2d
259 def from_reference(cls, ref_qconv, output_scale, output_zero_point): member in ConvReLU3d
|
| H A D | bn_relu.py | 56 def from_reference(cls, bn_relu, output_scale, output_zero_point): member in BNReLU2d
104 def from_reference(cls, bn_relu, output_scale, output_zero_point): member in BNReLU3d
|
| H A D | conv_add.py | 76 def from_reference(cls, ref_qconv, output_scale, output_zero_point): member in ConvAdd2d
144 def from_reference(cls, ref_qconv, output_scale, output_zero_point): member in ConvAddReLU2d
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/ |
| H A D | rnn.py | 665 def from_reference(cls, ref_mod): member in LSTM
930 def from_reference(cls, ref_mod): member in GRU
1104 def from_reference(cls, ref_mod): member in RNNCellBase
|
| H A D | linear.py | 150 def from_reference(cls, ref_qlinear): member in Linear
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/quantized/dynamic/modules/ |
| H A D | linear_relu.py | 59 def from_reference(cls, ref_qlinear_relu): member in LinearReLU
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/experimental/ |
| H A D | linear.py | 165 def from_reference( # type: ignore[override] member in LinearAPoT
|