Searched full:quantized_lstm (Results 1 – 17 of 17) sorted by relevance
| /external/pytorch/torch/ao/quantization/fx/ |
| D | lstm_utils.py | 162 quantized_lstm = torch.ao.nn.quantized.LSTM(
172 for i, layer in enumerate(quantized_lstm.layers):
202 return quantized_lstm
|
| /external/pytorch/aten/src/ATen/native/ |
| D | RNN.cpp | 1726 // BC wrappers for quantized_lstm
1742 "torch.quantized_lstm with List[Tensor] for parameters is " in quantized_lstm_input_legacy()
1792 "torch.quantized_lstm with List[Tensor] for parameters is " in quantized_lstm_data_legacy()
1908 …TORCH_SELECTIVE_SCHEMA("aten::quantized_lstm.input(Tensor input, Tensor[] hx, __torch__.torch.clas… in TORCH_LIBRARY_FRAGMENT()
1910 …TORCH_SELECTIVE_SCHEMA("aten::quantized_lstm.data(Tensor data, Tensor batch_sizes, Tensor[] hx, __… in TORCH_LIBRARY_FRAGMENT()
1912 …TORCH_SELECTIVE_SCHEMA("aten::quantized_lstm.input_legacy(Tensor input, Tensor[] hx, Tensor[] para… in TORCH_LIBRARY_FRAGMENT()
1914 …TORCH_SELECTIVE_SCHEMA("aten::quantized_lstm.data_legacy(Tensor data, Tensor batch_sizes, Tensor[]… in TORCH_LIBRARY_FRAGMENT()
1936 m.impl(TORCH_SELECTIVE_NAME("aten::quantized_lstm.input"), TORCH_FN(quantized_lstm_input)); in TORCH_LIBRARY_IMPL()
1937 m.impl(TORCH_SELECTIVE_NAME("aten::quantized_lstm.data"), TORCH_FN(quantized_lstm_data)); in TORCH_LIBRARY_IMPL()
1938 …m.impl(TORCH_SELECTIVE_NAME("aten::quantized_lstm.input_legacy"), TORCH_FN(quantized_lstm_input_le… in TORCH_LIBRARY_IMPL()
[all …]
|
| D | native_functions.yaml | 7812 # - func: quantized_lstm(Tensor input, Tensor[] hx, Tensor[] params, bool has_biases, int num_layer…
7815 # - func: quantized_lstm.data(Tensor data, Tensor batch_sizes, Tensor[] hx, Tensor[] params, bool h…
|
| /external/android-nn-driver/ |
| D | NnapiSupport.txt | 67 QUANTIZED_LSTM (QUANT8_ASYMM)
|
| /external/pytorch/test/expect/ |
| D | HasDecompTest.test_has_decomposition.expect | 1050 aten::quantized_lstm.data
1051 aten::quantized_lstm.data_legacy
1052 aten::quantized_lstm.input
1053 aten::quantized_lstm.input_legacy
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | acceleration_test_list.cc | 343 # quantized_lstm op test
|
| /external/pytorch/test/mobile/model_test/ |
| D | model_ops.yaml | 289 aten::quantized_lstm.data: 13
|
| D | coverage.yaml | 484 - aten::quantized_lstm.input
1069 aten::quantized_lstm.data: 2
|
| /external/android-nn-driver/test/1.3/ |
| D | QLstm.cpp | 447 // make the QUANTIZED_LSTM operation in QLstmTestImpl()
449 model.main.operations[0].type = HalPolicy::OperationType::QUANTIZED_LSTM; in QLstmTestImpl()
|
| /external/pytorch/torch/ao/nn/quantized/dynamic/modules/ |
| D | rnn.py | 561 result = torch.quantized_lstm(
575 result = torch.quantized_lstm(
|
| /external/android-nn-driver/1.3/ |
| D | HalPolicy.cpp | 124 case V1_3::OperationType::QUANTIZED_LSTM: in ConvertOperation()
|
| /external/pytorch/torch/ |
| D | __init__.py | 2154 quantized_lstm = ops.aten.quantized_lstm variable
|
| /external/android-nn-driver/test/ |
| D | Lstm.hpp | 2358 …// This replicates android/frameworks/ml/nn/runtime/test/generated/vts_models/quantized_lstm.model… in QuantizedLstm()
2359 …// with values from android/frameworks/ml/nn/runtime/test/generated/examples/quantized_lstm.exampl… in QuantizedLstm()
|
| /external/pytorch/test/quantization/core/ |
| D | test_quantized_module.py | 1743 result = torch.quantized_lstm(x, (h, c),
|
| D | test_quantized_op.py | 3532 result_dynamic = torch.quantized_lstm(Xq.dequantize(),
3554 result_dynamic = torch.quantized_lstm(Xq.dequantize(),
3608 … self.assertEqual(result_ref[0], result_dynamic[0], msg="torch.quantized_lstm results are off")
|
| /external/armnn/shim/sl/canonical/ |
| D | Converter.cpp | 125 case OperationType::QUANTIZED_LSTM: in ConvertOperation()
|
| /external/pytorch/torch/csrc/jit/frontend/ |
| D | ir_emitter.cpp | 3139 // Tensor here, then the quantized_lstm.legacy overload can kick in in in emitSingleAssignment()
|