| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/qat/modules/ |
| H A D | conv_fused.py | 41 _FLOAT_MODULE = MOD variable in _ConvBnNd
456 _FLOAT_MODULE = nni.ConvBn1d variable in ConvBn1d
523 _FLOAT_MODULE = nni.ConvBnReLU1d # type: ignore[assignment] variable in ConvBnReLU1d
588 _FLOAT_MODULE = nni.ConvReLU1d # type: ignore[assignment] variable in ConvReLU1d
651 _FLOAT_MODULE = nni.ConvBn2d variable in ConvBn2d
720 _FLOAT_MODULE = nni.ConvBnReLU2d # type: ignore[assignment] variable in ConvBnReLU2d
785 _FLOAT_MODULE = nni.ConvReLU2d # type: ignore[assignment] variable in ConvReLU2d
848 _FLOAT_MODULE = nni.ConvBn3d variable in ConvBn3d
916 _FLOAT_MODULE = nni.ConvBnReLU3d # type: ignore[assignment] variable in ConvBnReLU3d
983 _FLOAT_MODULE = nni.ConvReLU3d # type: ignore[assignment] variable in ConvReLU3d
|
| H A D | linear_relu.py | 31 _FLOAT_MODULE = nni.LinearReLU # type: ignore[assignment] variable in LinearReLU
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/ |
| H A D | conv.py | 50 _FLOAT_MODULE = nn.Conv1d variable in Conv1d
135 _FLOAT_MODULE = nn.Conv2d variable in Conv2d
219 _FLOAT_MODULE = nn.Conv3d variable in Conv3d
311 _FLOAT_MODULE = nn.ConvTranspose1d variable in ConvTranspose1d
393 _FLOAT_MODULE = nn.ConvTranspose2d variable in ConvTranspose2d
475 _FLOAT_MODULE = nn.ConvTranspose3d variable in ConvTranspose3d
|
| H A D | rnn.py | 92 _FLOAT_MODULE = nn.RNNBase variable in RNNBase
524 _FLOAT_MODULE = nn.LSTM variable in LSTM
808 _FLOAT_MODULE = nn.GRU variable in GRU
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/qat/modules/ |
| H A D | conv.py | 17 _FLOAT_MODULE = MOD variable in _ConvNd
139 _FLOAT_MODULE = nn.Conv1d variable in Conv1d
200 _FLOAT_MODULE = nn.Conv2d variable in Conv2d
264 _FLOAT_MODULE = nn.Conv3d variable in Conv3d
|
| H A D | embedding_ops.py | 26 _FLOAT_MODULE = nn.Embedding variable in Embedding
140 _FLOAT_MODULE = nn.EmbeddingBag variable in EmbeddingBag
|
| H A D | linear.py | 31 _FLOAT_MODULE = nn.Linear variable in Linear
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
| H A D | conv.py | 389 _FLOAT_MODULE = nn.Conv1d variable in Conv1d
521 _FLOAT_MODULE = nn.Conv2d variable in Conv2d
650 _FLOAT_MODULE = nn.Conv3d variable in Conv3d
747 _FLOAT_MODULE = MOD variable in _ConvTransposeNd
917 _FLOAT_MODULE = nn.ConvTranspose1d variable in ConvTranspose1d
1040 _FLOAT_MODULE = nn.ConvTranspose2d variable in ConvTranspose2d
1165 _FLOAT_MODULE = nn.ConvTranspose3d variable in ConvTranspose3d
|
| H A D | rnn.py | 35 _FLOAT_MODULE = torch.ao.nn.quantizable.LSTM # type: ignore[assignment] variable in LSTM
|
| H A D | linear.py | 149 _FLOAT_MODULE = (nn.Linear, nn.modules.linear.NonDynamicallyQuantizableLinear) variable in Linear
|
| H A D | activation.py | 229 _FLOAT_MODULE = torch.ao.nn.quantizable.MultiheadAttention variable in MultiheadAttention
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/quantized/modules/ |
| H A D | linear_relu.py | 33 _FLOAT_MODULE = nni.LinearReLU # type: ignore[assignment] variable in LinearReLU
73 _FLOAT_MODULE = nni.LinearLeakyReLU # type: ignore[assignment] variable in LinearLeakyReLU
148 _FLOAT_MODULE = nni.LinearTanh # type: ignore[assignment] variable in LinearTanh
|
| H A D | conv_relu.py | 31 _FLOAT_MODULE = torch.ao.nn.intrinsic.ConvReLU1d # type: ignore[assignment] variable in ConvReLU1d
112 _FLOAT_MODULE = torch.ao.nn.intrinsic.ConvReLU2d # type: ignore[assignment] variable in ConvReLU2d
193 _FLOAT_MODULE = torch.ao.nn.intrinsic.ConvReLU3d # type: ignore[assignment] variable in ConvReLU3d
|
| H A D | bn_relu.py | 22 _FLOAT_MODULE = torch.ao.nn.intrinsic.BNReLU2d variable in BNReLU2d
70 _FLOAT_MODULE = torch.ao.nn.intrinsic.BNReLU3d variable in BNReLU3d
|
| H A D | conv_add.py | 22 _FLOAT_MODULE = torch.ao.nn.intrinsic.ConvAdd2d # type: ignore[assignment] variable in ConvAdd2d
90 _FLOAT_MODULE = torch.ao.nn.intrinsic.ConvAddReLU2d # type: ignore[assignment] variable in ConvAddReLU2d
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantizable/modules/ |
| H A D | rnn.py | 36 _FLOAT_MODULE = torch.nn.LSTMCell variable in LSTMCell
332 _FLOAT_MODULE = torch.nn.LSTM variable in LSTM
|
| H A D | activation.py | 15 _FLOAT_MODULE = nn.MultiheadAttention variable in MultiheadAttention
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/quantized/dynamic/modules/ |
| H A D | linear_relu.py | 30 _FLOAT_MODULE = nni.LinearReLU # type: ignore[assignment] variable in LinearReLU
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/dynamic/ |
| H A D | linear.py | 23 _FLOAT_MODULE = torch.nn.Linear variable in Linear
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/ |
| H A D | linear.py | 108 _FLOAT_MODULE = torch.nn.Linear variable in Linear
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_qat.py | 230 _FLOAT_MODULE = torch.ao.nn.intrinsic.ConvBn2d variable in _ReferenceConvBn2d
|