Searched defs:in_cpu3 (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_api_test.cpp | 5814 const auto in_cpu3 = at::rand({3, 9, 331, 193}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5826 const auto in_cpu3 = at::rand({3, 9, 331, 193}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5839 const auto in_cpu3 = at::rand({221, 3, 9, 193}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5858 const auto in_cpu3 = at::rand({139, 3, 9, 193}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5877 const auto in_cpu3 = at::rand({1, 1, 221, 193}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5931 const auto in_cpu3 = at::rand({331, 9, 193, 3}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5950 const auto in_cpu3 = at::rand({9, 331, 193, 3}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5969 const auto in_cpu3 = at::rand({9, 193, 331, 3}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5988 const auto in_cpu3 = at::rand({9, 193, 3, 331}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
6008 const auto in_cpu3 = at::rand({3, 9, 221, 193}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
[all …]
|
| H A D | vulkan_quantized_api_test.cpp | 1232 const auto in_cpu3 = in TEST_F() local
1284 const auto in_cpu3 = in TEST_F() local
1336 const auto in_cpu3 = in TEST_F() local
1388 const auto in_cpu3 = in TEST_F() local
|