Searched defs:other_cpu (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_api_test.cpp | 433 const auto other_cpu = at::rand(other_shape, at::device(at::kCPU).dtype(at::kFloat)); in test_add() local
452 auto other_cpu = at::rand({2, 4, 4, 5}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
500 const auto other_cpu = in test_add_other_cpu_int() local
529 const auto other_cpu = in TEST_F() local
575 const auto other_cpu = in TEST_F() local
2217 const auto other_cpu = at::rand(other_shape, at::device(at::kCPU).dtype(at::kFloat)) + 0.01; in test_div() local
3626 const auto other_cpu = at::rand(other_shape, at::device(at::kCPU).dtype(at::kFloat)); in test_mul() local
3831 const auto other_cpu = at::rand(other_shape, at::device(at::kCPU).dtype(at::kFloat)); in test_pow() local
3885 const auto other_cpu = at::rand(other_shape, at::device(at::kCPU).dtype(at::kFloat)); in test_pow_() local
3967 const auto other_cpu = at::rand(other, at::device(at::kCPU).dtype(at::kFloat)); in test_pow_scalar_tensor() local
[all …]
|