Searched refs:test_cpu_to_vulkan_and_dequantize (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_quantized_api_test.cpp | 583 void test_cpu_to_vulkan_and_dequantize( in test_cpu_to_vulkan_and_dequantize() function
619 test_cpu_to_vulkan_and_dequantize(tensor_shape, scale, zero_point, dtype); in test_cpu_to_vulkan_and_dequantize_random()
624 test_cpu_to_vulkan_and_dequantize({1, 1, 1, 1}, 0.13, 21, dtype); in TEST_F()
625 test_cpu_to_vulkan_and_dequantize({1, 1, 1, 4}, 0.3, 87, dtype); in TEST_F()
626 test_cpu_to_vulkan_and_dequantize({1, 1, 4, 1}, 0.2, 120, dtype); in TEST_F()
627 test_cpu_to_vulkan_and_dequantize({1, 1, 7, 7}, 0.3, 87, dtype); in TEST_F()
628 test_cpu_to_vulkan_and_dequantize({1, 1, 8, 8}, 0.1, 10, dtype); in TEST_F()
629 test_cpu_to_vulkan_and_dequantize({3, 5, 8, 8}, 0.04, 97, dtype); in TEST_F()
630 test_cpu_to_vulkan_and_dequantize({1, 1, 11, 17}, 0.07, 15, dtype); in TEST_F()
631 test_cpu_to_vulkan_and_dequantize({1, 1, 12, 17}, 0.1, 10, dtype); in TEST_F()
[all …]
|