Searched defs:in_gpu (Results 1 – 4 of 4) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_reduction_sycl.cpp | 42 TensorMap<Tensor<float, 2> > in_gpu(gpu_in_data, tensorRange); in test_full_reductions_sycl() local
77 TensorMap<Tensor<float, 3> > in_gpu(gpu_in_data, tensorRange); in test_first_dim_reductions_sycl() local
115 TensorMap<Tensor<float, 3> > in_gpu(gpu_in_data, tensorRange); in test_last_dim_reductions_sycl() local
|
| D | cxx11_tensor_complex_cuda.cu | 94 TensorMap<Tensor<std::complex<float>, 2> > in_gpu(gpu_in_ptr, num_rows, num_cols); in test_cuda_sum_reductions() local
131 TensorMap<Tensor<std::complex<float>, 2> > in_gpu(gpu_in_ptr, num_rows, num_cols); in test_cuda_product_reductions() local
|
| D | cxx11_tensor_reduction_cuda.cu | 43 TensorMap<Tensor<Type, 2, DataLayout> > in_gpu(gpu_in_ptr, num_rows, num_cols); in test_full_reductions() local
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 127 const Tensor& in_gpu = test_case->ins.back(); in MakeReductionTestCase() local
168 const Tensor& in_gpu = test_case->ins.back(); in MakeGatherTestCase() local
|