Searched refs:gpu_in_ptr (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_complex_cuda.cu | 90 …std::complex<float>* gpu_in_ptr = static_cast<std::complex<float>*>(gpu_device.allocate(in_bytes)); in test_cuda_sum_reductions() local
92 gpu_device.memcpyHostToDevice(gpu_in_ptr, in.data(), in_bytes); in test_cuda_sum_reductions()
94 TensorMap<Tensor<std::complex<float>, 2> > in_gpu(gpu_in_ptr, num_rows, num_cols); in test_cuda_sum_reductions()
106 gpu_device.deallocate(gpu_in_ptr); in test_cuda_sum_reductions()
127 …std::complex<float>* gpu_in_ptr = static_cast<std::complex<float>*>(gpu_device.allocate(in_bytes)); in test_cuda_product_reductions() local
129 gpu_device.memcpyHostToDevice(gpu_in_ptr, in.data(), in_bytes); in test_cuda_product_reductions()
131 TensorMap<Tensor<std::complex<float>, 2> > in_gpu(gpu_in_ptr, num_rows, num_cols); in test_cuda_product_reductions()
143 gpu_device.deallocate(gpu_in_ptr); in test_cuda_product_reductions()
|
| D | cxx11_tensor_reduction_cuda.cu | 39 Type* gpu_in_ptr = static_cast<Type*>(gpu_device.allocate(in_bytes)); in test_full_reductions() local
41 gpu_device.memcpyHostToDevice(gpu_in_ptr, in.data(), in_bytes); in test_full_reductions()
43 TensorMap<Tensor<Type, 2, DataLayout> > in_gpu(gpu_in_ptr, num_rows, num_cols); in test_full_reductions()
55 gpu_device.deallocate(gpu_in_ptr); in test_full_reductions()
|