Searched refs:test_quantize_per_tensor_and_vulkan_to_cpu (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/test/ |
| D | vulkan_quantized_api_test.cpp | 729 void test_quantize_per_tensor_and_vulkan_to_cpu( in test_quantize_per_tensor_and_vulkan_to_cpu() function
773 test_quantize_per_tensor_and_vulkan_to_cpu( in test_quantize_per_tensor_and_vulkan_to_cpu_random()
780 test_quantize_per_tensor_and_vulkan_to_cpu({1, 1, 1, 1}, 0.13, 21, dtype); in TEST_F()
781 test_quantize_per_tensor_and_vulkan_to_cpu({1, 1, 1, 4}, 0.3, 87, dtype); in TEST_F()
782 test_quantize_per_tensor_and_vulkan_to_cpu({1, 1, 4, 1}, 0.2, 120, dtype); in TEST_F()
783 test_quantize_per_tensor_and_vulkan_to_cpu({1, 1, 7, 7}, 0.3, 87, dtype); in TEST_F()
784 test_quantize_per_tensor_and_vulkan_to_cpu({1, 1, 8, 8}, 0.1, 10, dtype); in TEST_F()
785 test_quantize_per_tensor_and_vulkan_to_cpu({3, 5, 8, 8}, 0.04, 97, dtype); in TEST_F()
786 test_quantize_per_tensor_and_vulkan_to_cpu({1, 1, 11, 17}, 0.07, 15, dtype); in TEST_F()
787 test_quantize_per_tensor_and_vulkan_to_cpu({1, 1, 12, 17}, 0.1, 10, dtype); in TEST_F()
[all …]
|