Searched refs:_get_random_tensor_and_q_params (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_op.py | 125 def _get_random_tensor_and_q_params(shapes, rand_scale, torch_type): function
582 _get_random_tensor_and_q_params(shapes, 1.0, torch_type)
797 _get_random_tensor_and_q_params(shapes, 2.0, torch_type)
1904 _get_random_tensor_and_q_params(shapes, 1.0, zero_point)
1972 _get_random_tensor_and_q_params(shapes, 1.0, zero_point)
2051 _get_random_tensor_and_q_params(shapes, 1.0, zero_point)
2122 X, scale, zp = _get_random_tensor_and_q_params(shape, 1.0, dtype)
2520 _get_random_tensor_and_q_params(shapes, 1.0, torch_type)
2621 _get_random_tensor_and_q_params(shapes, 1.0, torch_type)
2687 _get_random_tensor_and_q_params(shapes, 1.0, torch_type)
[all …]
|