Searched refs:gpu_out_ptr (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_complex_cuda.cu | 91 …std::complex<float>* gpu_out_ptr = static_cast<std::complex<float>*>(gpu_device.allocate(out_bytes… in test_cuda_sum_reductions() local
95 TensorMap<Tensor<std::complex<float>, 0> > out_gpu(gpu_out_ptr); in test_cuda_sum_reductions()
100 gpu_device.memcpyDeviceToHost(full_redux_gpu.data(), gpu_out_ptr, out_bytes); in test_cuda_sum_reductions()
107 gpu_device.deallocate(gpu_out_ptr); in test_cuda_sum_reductions()
128 …std::complex<float>* gpu_out_ptr = static_cast<std::complex<float>*>(gpu_device.allocate(out_bytes… in test_cuda_product_reductions() local
132 TensorMap<Tensor<std::complex<float>, 0> > out_gpu(gpu_out_ptr); in test_cuda_product_reductions()
137 gpu_device.memcpyDeviceToHost(full_redux_gpu.data(), gpu_out_ptr, out_bytes); in test_cuda_product_reductions()
144 gpu_device.deallocate(gpu_out_ptr); in test_cuda_product_reductions()
|
| D | cxx11_tensor_reduction_cuda.cu | 40 Type* gpu_out_ptr = static_cast<Type*>(gpu_device.allocate(out_bytes)); in test_full_reductions() local
44 TensorMap<Tensor<Type, 0, DataLayout> > out_gpu(gpu_out_ptr); in test_full_reductions()
49 gpu_device.memcpyDeviceToHost(full_redux_gpu.data(), gpu_out_ptr, out_bytes); in test_full_reductions()
56 gpu_device.deallocate(gpu_out_ptr); in test_full_reductions()
|