Searched refs:d_t_left (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_contract_cuda.cu | 46 float* d_t_left; in test_cuda_contraction() local
50 cudaMalloc((void**)(&d_t_left), t_left_bytes); in test_cuda_contraction()
54 cudaMemcpy(d_t_left, t_left.data(), t_left_bytes, cudaMemcpyHostToDevice); in test_cuda_contraction()
61 gpu_t_left(d_t_left, Eigen::array<int, 2>(m_size, k_size)); in test_cuda_contraction()
84 cudaFree((void*)d_t_left); in test_cuda_contraction()
110 float* d_t_left; in test_scalar() local
114 cudaMalloc((void**)(&d_t_left), t_left_bytes); in test_scalar()
118 cudaMemcpy(d_t_left, t_left.data(), t_left_bytes, cudaMemcpyHostToDevice); in test_scalar()
125 gpu_t_left(d_t_left, m_size, k_size); in test_scalar()
142 cudaFree((void*)d_t_left); in test_scalar()
|
| D | cxx11_tensor_cuda.cu | 273 float* d_t_left; in test_cuda_contraction() local
277 cudaMalloc((void**)(&d_t_left), t_left_bytes); in test_cuda_contraction()
281 cudaMemcpy(d_t_left, t_left.data(), t_left_bytes, cudaMemcpyHostToDevice); in test_cuda_contraction()
287 Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout> > gpu_t_left(d_t_left, 6, 50, 3, 31); in test_cuda_contraction()
313 cudaFree(d_t_left); in test_cuda_contraction()
|