Searched refs:d_t_right (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_contract_cuda.cu | 47 float* d_t_right; in test_cuda_contraction() local
51 cudaMalloc((void**)(&d_t_right), t_right_bytes); in test_cuda_contraction()
55 cudaMemcpy(d_t_right, t_right.data(), t_right_bytes, cudaMemcpyHostToDevice); in test_cuda_contraction()
63 gpu_t_right(d_t_right, Eigen::array<int, 2>(k_size, n_size)); in test_cuda_contraction()
85 cudaFree((void*)d_t_right); in test_cuda_contraction()
111 float* d_t_right; in test_scalar() local
115 cudaMalloc((void**)(&d_t_right), t_right_bytes); in test_scalar()
119 cudaMemcpy(d_t_right, t_right.data(), t_right_bytes, cudaMemcpyHostToDevice); in test_scalar()
127 gpu_t_right(d_t_right, k_size, n_size); in test_scalar()
143 cudaFree((void*)d_t_right); in test_scalar()
|
| D | cxx11_tensor_cuda.cu | 274 float* d_t_right; in test_cuda_contraction() local
278 cudaMalloc((void**)(&d_t_right), t_right_bytes); in test_cuda_contraction()
282 cudaMemcpy(d_t_right, t_right.data(), t_right_bytes, cudaMemcpyHostToDevice); in test_cuda_contraction()
288 Eigen::TensorMap<Eigen::Tensor<float, 5, DataLayout> > gpu_t_right(d_t_right, 3, 31, 7, 20, 1); in test_cuda_contraction()
314 cudaFree(d_t_right); in test_cuda_contraction()
|