Searched refs:gpu_t_left (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_contract_cuda.cu | 61 gpu_t_left(d_t_left, Eigen::array<int, 2>(m_size, k_size)); in test_cuda_contraction()
68 gpu_t_result.device(gpu_device) = gpu_t_left.contract(gpu_t_right, dims); in test_cuda_contraction()
125 gpu_t_left(d_t_left, m_size, k_size); in test_scalar() local
131 gpu_t_result.device(gpu_device) = gpu_t_left.contract(gpu_t_right, dims); in test_scalar()
|
| D | cxx11_tensor_cuda.cu | 287 Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout> > gpu_t_left(d_t_left, 6, 50, 3, 31); in test_cuda_contraction() local
302 gpu_t_result.device(gpu_device) = gpu_t_left.contract(gpu_t_right, dims); in test_cuda_contraction()
|