Home
last modified time | relevance | path

Searched defs:out_cpu (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp188 const auto out_cpu = at::slice(in_cpu, dim, start, end, step); in slice_test() local
221 const auto out_cpu = at::clone(in_cpu, optional_memory_format); in clone_test() local
420 const auto out_cpu = at::adaptive_avg_pool2d(in_cpu, {3, 3}); in TEST_F() local
438 const auto out_cpu = at::add(in_cpu, other_cpu, alpha); in test_add() local
506 const auto out_cpu = at::add(in_cpu, other_cpu, alpha); in test_add_other_cpu_int() local
713 const auto out_cpu = at::addmm(bias_cpu, m1_cpu, m2_cpu, beta, alpha); in TEST_F() local
733 const auto out_cpu = at::addmm(bias_cpu, m1_cpu, m2_cpu, beta, alpha); in TEST_F() local
753 const auto out_cpu = at::addmm(bias_cpu, m1_cpu, m2_cpu, beta, alpha); in TEST_F() local
780 const auto out_cpu = at::avg_pool2d(in_cpu, {5, 3}, {1, 2}, {2, 0}, true); in TEST_F() local
1001 const auto out_cpu = at::baddbmm(bias_cpu, m1_cpu, m2_cpu, beta, alpha); in test_baddbmm() local
[all …]
H A Dvulkan_quantized_api_test.cpp709 const auto out_cpu = at::quantize_per_tensor( in TEST_F() local
886 const auto out_cpu = at::quantize_per_tensor( in TEST_F() local
1170 const auto out_cpu = at::quantize_per_tensor( in TEST_F() local
1216 const auto out_cpu = at::quantize_per_tensor( in TEST_F() local
1268 const auto out_cpu = at::quantize_per_tensor( in TEST_F() local
1320 const auto out_cpu = at::quantize_per_tensor( in TEST_F() local
1372 const auto out_cpu = at::quantize_per_tensor( in TEST_F() local
1421 const auto out_cpu = at::quantize_per_tensor( in TEST_F() local
1842 const at::Tensor out_cpu = at::dequantize(out_cpu_quant[0].toTensor()); in test_quantized_conv_transpose2d() local
1924 const auto out_cpu = at::quantize_per_tensor( in TEST_F() local
[all …]
H A Dxnnpack_test.cpp74 const auto out_cpu = at::linear(input_cpu, weight, bias); in TEST() local
88 const auto out_cpu = in TEST() local
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_math_sycl.cpp40 Tensor<DataType, 3, DataLayout, IndexType> out_cpu(tensorRange); in test_tanh_sycl() local
70 Tensor<DataType, 3, DataLayout, IndexType> out_cpu(tensorRange); in test_sigmoid_sycl() local
/aosp_15_r20/external/tensorflow/tensorflow/core/nccl/
H A Dnccl_manager_test.cc268 Tensor out_cpu(data_type_, out_gpu.shape()); in VerifyResults() local