Searched refs:test_cpu_to_vulkan_and_vulkan_to_cpu (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/test/ |
| D | vulkan_quantized_api_test.cpp | 461 void test_cpu_to_vulkan_and_vulkan_to_cpu( in test_cpu_to_vulkan_and_vulkan_to_cpu() function
499 test_cpu_to_vulkan_and_vulkan_to_cpu(tensor_shape, scale, zero_point, dtype); in test_cpu_to_vulkan_and_vulkan_to_cpu_random()
505 test_cpu_to_vulkan_and_vulkan_to_cpu({1, 1, 1, 1}, 0.13, 21, dtype); in TEST_F()
506 test_cpu_to_vulkan_and_vulkan_to_cpu({1, 1, 1, 4}, 0.3, 87, dtype); in TEST_F()
507 test_cpu_to_vulkan_and_vulkan_to_cpu({1, 1, 4, 1}, 0.2, 120, dtype); in TEST_F()
508 test_cpu_to_vulkan_and_vulkan_to_cpu({1, 1, 7, 7}, 0.3, 87, dtype); in TEST_F()
509 test_cpu_to_vulkan_and_vulkan_to_cpu({1, 1, 8, 8}, 0.1, 10, dtype); in TEST_F()
510 test_cpu_to_vulkan_and_vulkan_to_cpu({3, 5, 8, 8}, 0.04, 97, dtype); in TEST_F()
511 test_cpu_to_vulkan_and_vulkan_to_cpu({1, 1, 11, 17}, 0.07, 15, dtype); in TEST_F()
512 test_cpu_to_vulkan_and_vulkan_to_cpu({1, 1, 12, 17}, 0.1, 10, dtype); in TEST_F()
[all …]
|