Searched refs: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
46 out_gpu.device(sycl_device) = in_gpu.sum(); in test_full_reductions_sycl()
77 TensorMap<Tensor<float, 3> > in_gpu(gpu_in_data, tensorRange); in test_first_dim_reductions_sycl() local
81 out_gpu.device(sycl_device) = in_gpu.sum(red_axis); in test_first_dim_reductions_sycl()
115 TensorMap<Tensor<float, 3> > in_gpu(gpu_in_data, tensorRange); in test_last_dim_reductions_sycl() local
119 out_gpu.device(sycl_device) = in_gpu.sum(red_axis); in test_last_dim_reductions_sycl()
|
| 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
97 out_gpu.device(gpu_device) = in_gpu.sum(); in test_cuda_sum_reductions()
131 TensorMap<Tensor<std::complex<float>, 2> > in_gpu(gpu_in_ptr, num_rows, num_cols); in test_cuda_product_reductions() local
134 out_gpu.device(gpu_device) = in_gpu.prod(); in test_cuda_product_reductions()
|
| 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
46 out_gpu.device(gpu_device) = in_gpu.sum(); in test_full_reductions()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 127 const Tensor& in_gpu = test_case->ins.back(); in MakeReductionTestCase() local
128 auto in_gpu_mem = AsDeviceMemory(in_gpu.flat<Scalar>().data()); in MakeReductionTestCase()
168 const Tensor& in_gpu = test_case->ins.back(); in MakeGatherTestCase() local
169 auto in_gpu_mem = AsDeviceMemory(in_gpu.flat<Scalar>().data()); in MakeGatherTestCase()
|