Home
last modified time | relevance | path

Searched defs:in_cpu2 (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp5813 const auto in_cpu2 = at::rand({3, 9, 112, 193}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5825 const auto in_cpu2 = at::rand({9, 112, 193}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5838 const auto in_cpu2 = at::rand({221, 3, 9, 193}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5857 const auto in_cpu2 = at::rand({117, 3, 9, 193}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5876 const auto in_cpu2 = at::rand({1, 1, 221, 193}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5912 const auto in_cpu2 = at::rand({3, 7, 221, 193}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5930 const auto in_cpu2 = at::rand({113, 9, 193, 3}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5949 const auto in_cpu2 = at::rand({9, 113, 193, 3}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5968 const auto in_cpu2 = at::rand({9, 193, 113, 3}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
5987 const auto in_cpu2 = at::rand({9, 193, 3, 113}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
[all …]
H A Dvulkan_quantized_api_test.cpp1163 const auto in_cpu2 = in TEST_F() local
1209 const auto in_cpu2 = in TEST_F() local
1261 const auto in_cpu2 = in TEST_F() local
1313 const auto in_cpu2 = in TEST_F() local
1365 const auto in_cpu2 = in TEST_F() local
1413 const auto in_cpu2 = in TEST_F() local
1916 const auto in_cpu2 = (r3 - r4) * in TEST_F() local
1960 const auto in_cpu2 = in TEST_F() local
2012 const auto in_cpu2 = (r3 - r4) * in TEST_F() local
2065 const auto in_cpu2 = in TEST_F() local