Searched refs:_construct_tensor_for_quantization_test (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_onnx_onnxruntime.py | 93 def _construct_tensor_for_quantization_test( function
13140 _construct_tensor_for_quantization_test((3, 4))
13146 input = _construct_tensor_for_quantization_test((4, 4), offset=-8)
13215 _construct_tensor_for_quantization_test((2, 4, 3, 3), max_val=2)
13221 input = _construct_tensor_for_quantization_test(
13248 _construct_tensor_for_quantization_test((2, 4, 3, 3), max_val=2)
13254 input = _construct_tensor_for_quantization_test(
13282 _construct_tensor_for_quantization_test((2, 4, 3, 3), max_val=2)
13288 input = _construct_tensor_for_quantization_test(
13316 _construct_tensor_for_quantization_test((2, 4), max_val=2)
[all …]
|