Searched refs:gpu_out_ptr (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_complex_gpu.cu | 88 …std::complex<float>* gpu_out_ptr = static_cast<std::complex<float>*>(gpu_device.allocate(out_bytes… in test_cuda_sum_reductions() local
92 TensorMap<Tensor<std::complex<float>, 0> > out_gpu(gpu_out_ptr); in test_cuda_sum_reductions()
97 gpu_device.memcpyDeviceToHost(full_redux_gpu.data(), gpu_out_ptr, out_bytes); in test_cuda_sum_reductions()
104 gpu_device.deallocate(gpu_out_ptr); in test_cuda_sum_reductions()
124 …std::complex<float>* gpu_out_ptr = static_cast<std::complex<float>*>(gpu_device.allocate(out_bytes… in test_cuda_mean_reductions() local
128 TensorMap<Tensor<std::complex<float>, 0> > out_gpu(gpu_out_ptr); in test_cuda_mean_reductions()
133 gpu_device.memcpyDeviceToHost(full_redux_gpu.data(), gpu_out_ptr, out_bytes); in test_cuda_mean_reductions()
140 gpu_device.deallocate(gpu_out_ptr); in test_cuda_mean_reductions()
160 …std::complex<float>* gpu_out_ptr = static_cast<std::complex<float>*>(gpu_device.allocate(out_bytes… in test_cuda_product_reductions() local
164 TensorMap<Tensor<std::complex<float>, 0> > out_gpu(gpu_out_ptr); in test_cuda_product_reductions()
[all …]
|
| D | cxx11_tensor_reduction_gpu.cu | 37 Type* gpu_out_ptr = static_cast<Type*>(gpu_device.allocate(out_bytes)); in test_full_reductions() local
41 TensorMap<Tensor<Type, 0, DataLayout> > out_gpu(gpu_out_ptr); in test_full_reductions()
46 gpu_device.memcpyDeviceToHost(full_redux_gpu.data(), gpu_out_ptr, out_bytes); in test_full_reductions()
53 gpu_device.deallocate(gpu_out_ptr); in test_full_reductions()
|