Searched defs:from_observed (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/torch/ao/nn/quantized/modules/ |
| D | rnn.py | 51 def from_observed(cls, other): member in LSTM
|
| D | activation.py | 245 def from_observed(cls, other): member in MultiheadAttention
|
| /external/pytorch/torch/ao/nn/quantizable/modules/ |
| D | activation.py | 266 def from_observed(cls, other): member in MultiheadAttention
|
| D | rnn.py | 492 def from_observed(cls, other): member in LSTM
|
| /external/pytorch/test/quantization/eager/ |
| D | test_quantize_eager_ptq.py | 895 def from_observed(cls, observed_module): member in TestQuantizeEagerPTQStatic.test_custom_module_class.QuantizedCustomModule
|
| /external/pytorch/test/quantization/fx/ |
| D | test_quantize_fx.py | 3173 def from_observed(cls, observed_module): member in TestQuantizeFx.test_custom_module_class.StaticQuantCustomModule
3190 def from_observed(cls, observed_module): member in TestQuantizeFx.test_custom_module_class.DynamicQuantCustomModule
3343 def from_observed(cls, observed_module): member in TestQuantizeFx.test_custom_module_class_input_has_multiple_users.StaticQuantCustomModule
3427 def from_observed(cls, observed_module): member in TestQuantizeFx.test_custom_module_class_input_has_duplicate_nodes.StaticQuantCustomModule
4946 def from_observed(cls, observed_lstm): member in TestQuantizeFx.test_static_lstm_with_custom_fixed_qparams.UserQuantizedLSTM
|