Home
last modified time | relevance | path

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 Dtest_quantized_op.py125 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 …]