Searched defs:in_gpu (Results 1 – 6 of 6) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_reduction_sycl.cpp | 47 TensorMap<Tensor<DataType, 2, DataLayout, IndexType>> in_gpu(gpu_in_data, in test_full_reductions_sum_sycl() local
92 TensorMap<data_tensor> in_gpu(gpu_in_data + offset, tensor_offset_range); in test_full_reductions_sum_with_offset_sycl() local
126 TensorMap<Tensor<DataType, 2, DataLayout, IndexType>> in_gpu(gpu_in_data, in test_full_reductions_max_sycl() local
170 TensorMap<data_tensor> in_gpu(gpu_in_data + offset, tensor_offset_range); in test_full_reductions_max_with_offset_sycl() local
221 TensorMap<Tensor<DataType, 2, DataLayout, IndexType>> in_gpu(gpu_in_data, in test_full_reductions_mean_sycl() local
300 TensorMap<data_tensor> in_gpu(gpu_in_data + offset, tensor_offset_range); in test_full_reductions_mean_with_offset_sycl() local
345 TensorMap<data_tensor> in_gpu(gpu_in_data, tensor_range); in test_full_reductions_mean_with_odd_offset_sycl() local
380 TensorMap<Tensor<DataType, 2, DataLayout, IndexType>> in_gpu(gpu_in_data, in test_full_reductions_min_sycl() local
426 TensorMap<data_tensor> in_gpu(gpu_in_data + offset, tensor_offset_range); in test_full_reductions_min_with_offset_sycl() local
465 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> in_gpu(gpu_in_data, in test_first_dim_reductions_max_sycl() local
[all …]
|
| D | cxx11_tensor_complex_gpu.cu | 91 TensorMap<Tensor<std::complex<float>, 2> > in_gpu(gpu_in_ptr, num_rows, num_cols); in test_cuda_sum_reductions() local
127 TensorMap<Tensor<std::complex<float>, 2> > in_gpu(gpu_in_ptr, num_rows, num_cols); in test_cuda_mean_reductions() local
163 TensorMap<Tensor<std::complex<float>, 2> > in_gpu(gpu_in_ptr, num_rows, num_cols); in test_cuda_product_reductions() local
|
| D | cxx11_tensor_reverse_sycl.cpp | 46 TensorMap<Tensor<DataType, 4, DataLayout, IndexType> > in_gpu(gpu_in_data, in test_simple_reverse() local
140 TensorMap<Tensor<DataType, 4, DataLayout, IndexType> > in_gpu(gpu_in_data, in test_expr_reverse() local
|
| D | cxx11_tensor_reduction_gpu.cu | 40 TensorMap<Tensor<Type, 2, DataLayout> > in_gpu(gpu_in_ptr, num_rows, num_cols); in test_full_reductions() local
|
| /external/eigen/test/ |
| D | gpu_common.h | 98 Input in_ref, in_gpu; in run_and_compare_to_gpu() local
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 155 const Tensor& in_gpu = test_case->ins.back(); in MakeReductionTestCase() local
196 const Tensor& in_gpu = test_case->ins.back(); in MakeGatherTestCase() local
232 const Tensor& in_gpu = test_case->ins.back(); in MakeBroadcastTestCase() local
|