Searched defs:in_cpu (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_api_test.cpp | 184 const auto in_cpu = at::rand(size, at::device(at::kCPU).dtype(at::kFloat)); in slice_test() local
217 const auto in_cpu = at::rand(size, at::device(at::kCPU).dtype(at::kFloat)); in clone_test() local
349 for (auto in_cpu : test_tensors) { in TEST_F() local
419 const auto in_cpu = at::rand({5, 7, 47, 31}, at::TensorOptions(at::kCPU).dtype(at::kFloat)); in TEST_F() local
432 const auto in_cpu = at::rand(input_shape, at::device(at::kCPU).dtype(at::kFloat)); in test_add() local
451 auto in_cpu = at::rand({2, 3, 4, 5}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
498 const auto in_cpu = in test_add_other_cpu_int() local
527 const auto in_cpu = at::rand({2,2,2}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
573 const auto in_cpu = in TEST_F() local
779 const auto in_cpu = at::rand({3, 19, 43, 79}, at::TensorOptions(at::kCPU).dtype(at::kFloat)); in TEST_F() local
[all …]
|
| H A D | vulkan_quantized_api_test.cpp | 297 at::Tensor cpu_to_vulkan(at::Tensor in_cpu) { in cpu_to_vulkan()
320 at::Tensor vulkan_to_cpu(at::Tensor vulkan, at::Tensor in_cpu) { in vulkan_to_cpu()
381 for (auto in_cpu : test_tensors) { in TEST_F() local
407 for (auto in_cpu : test_tensors) { in TEST_F() local
427 auto in_cpu = in TEST_F() local
467 auto in_cpu = produce_random_tensor(input_shape); in test_cpu_to_vulkan_and_vulkan_to_cpu() local
589 auto in_cpu = produce_random_tensor(input_shape); in test_cpu_to_vulkan_and_dequantize() local
702 const auto in_cpu = in TEST_F() local
879 const auto in_cpu = in TEST_F() local
1160 const auto in_cpu = in TEST_F() local
[all …]
|
| H A D | xnnpack_test.cpp | 86 const auto in_cpu = in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/nccl/ |
| H A D | nccl_manager_test.cc | 129 Tensor in_cpu(data_type_, shape); in MakeReductionTestCase() local
178 Tensor in_cpu(data_type_, in_shape); in MakeGatherTestCase() local
229 Tensor in_cpu(data_type_, shape); in MakeBroadcastTestCase() local
|