Searched refs:d_t_result (Results 1 – 3 of 3) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_contract_cuda.cu | 48 float* d_t_result; in test_cuda_contraction() local
52 cudaMalloc((void**)(&d_t_result), t_result_bytes); in test_cuda_contraction()
65 gpu_t_result(d_t_result, Eigen::array<int, 2>(m_size, n_size)); in test_cuda_contraction()
71 cudaMemcpy(t_result_gpu.data(), d_t_result, t_result_bytes, cudaMemcpyDeviceToHost); in test_cuda_contraction()
86 cudaFree((void*)d_t_result); in test_cuda_contraction()
112 float* d_t_result; in test_scalar() local
116 cudaMalloc((void**)(&d_t_result), t_result_bytes); in test_scalar()
129 gpu_t_result(d_t_result); in test_scalar()
134 cudaMemcpy(t_result_gpu.data(), d_t_result, t_result_bytes, cudaMemcpyDeviceToHost); in test_scalar()
144 cudaFree((void*)d_t_result); in test_scalar()
|
| D | cxx11_tensor_scan_cuda.cu | 39 float* d_t_result; in test_cuda_cumsum() local
42 cudaMalloc((void**)(&d_t_result), t_result_bytes); in test_cuda_cumsum()
52 gpu_t_result(d_t_result, Eigen::array<int, 3>(m_size, k_size, n_size)); in test_cuda_cumsum()
57 cudaMemcpy(t_result_gpu.data(), d_t_result, t_result_bytes, cudaMemcpyDeviceToHost); in test_cuda_cumsum()
71 cudaFree((void*)d_t_result); in test_cuda_cumsum()
|
| D | cxx11_tensor_cuda.cu | 275 float* d_t_result; in test_cuda_contraction() local
279 cudaMalloc((void**)(&d_t_result), t_result_bytes); in test_cuda_contraction()
289 Eigen::TensorMap<Eigen::Tensor<float, 5, DataLayout> > gpu_t_result(d_t_result, 6, 50, 7, 20, 1); in test_cuda_contraction()
304 cudaMemcpy(t_result.data(), d_t_result, t_result_bytes, cudaMemcpyDeviceToHost); in test_cuda_contraction()
315 cudaFree(d_t_result); in test_cuda_contraction()
|