Searched defs:out_gpu (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_reduction_sycl.cpp | 49 TensorMap<Tensor<DataType, 2, DataLayout, IndexType>> out_gpu(gpu_out_data, in test_full_reductions_sum_sycl() local
93 TensorMap<scalar_tensor> out_gpu(gpu_out_data); in test_full_reductions_sum_with_offset_sycl() local
128 TensorMap<Tensor<DataType, 0, DataLayout, IndexType>> out_gpu(gpu_out_data); in test_full_reductions_max_sycl() local
171 TensorMap<scalar_tensor> out_gpu(gpu_out_data); in test_full_reductions_max_with_offset_sycl() local
231 TensorMap<Tensor<DataType, 0, DataLayout, IndexType>> out_gpu(gpu_out_data); in test_full_reductions_mean_sycl() local
301 TensorMap<scalar_tensor> out_gpu(gpu_out_data); in test_full_reductions_mean_with_offset_sycl() local
346 TensorMap<scalar_tensor> out_gpu(gpu_out_data); in test_full_reductions_mean_with_odd_offset_sycl() local
382 TensorMap<Tensor<DataType, 0, DataLayout, IndexType>> out_gpu(gpu_out_data); in test_full_reductions_min_sycl() local
427 TensorMap<scalar_tensor> out_gpu(gpu_out_data); in test_full_reductions_min_with_offset_sycl() local
467 TensorMap<Tensor<DataType, 2, DataLayout, IndexType>> out_gpu( in test_first_dim_reductions_max_sycl() local
[all …]
|
| H A D | cxx11_tensor_complex_gpu.cu | 92 TensorMap<Tensor<std::complex<float>, 0> > out_gpu(gpu_out_ptr); in test_cuda_sum_reductions() local
128 TensorMap<Tensor<std::complex<float>, 0> > out_gpu(gpu_out_ptr); in test_cuda_mean_reductions() local
164 TensorMap<Tensor<std::complex<float>, 0> > out_gpu(gpu_out_ptr); in test_cuda_product_reductions() local
|
| H A D | cxx11_tensor_reduction_gpu.cu | 41 TensorMap<Tensor<Type, 0, DataLayout> > out_gpu(gpu_out_ptr); in test_full_reductions() local
|
| H A D | cxx11_tensor_reverse_sycl.cpp | 48 TensorMap<Tensor<DataType, 4, DataLayout, IndexType> > out_gpu(gpu_out_data, in test_simple_reverse() local
|
| H A D | cxx11_tensor_morphing_sycl.cpp | 334 TensorMI64 out_gpu(out_gpu_data, tensor_range); in run_eigen() local
|
| /aosp_15_r20/external/eigen/test/ |
| H A D | gpu_common.h | 99 Output out_ref, out_gpu; in run_and_compare_to_gpu() local
|
| H A D | gpu_basic.cu | 384 Output out_ref, out_gpu; in test_with_infs_nans() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/nccl/ |
| H A D | nccl_manager_test.cc | 267 const Tensor& out_gpu = test_case->outs[global_rank]; in VerifyResults() local
|