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 …]
|