Searched refs:gpu_t_right (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_contract_cuda.cu | 63 gpu_t_right(d_t_right, Eigen::array<int, 2>(k_size, n_size)); in test_cuda_contraction()
68 gpu_t_result.device(gpu_device) = gpu_t_left.contract(gpu_t_right, dims); in test_cuda_contraction()
127 gpu_t_right(d_t_right, k_size, n_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 | 288 Eigen::TensorMap<Eigen::Tensor<float, 5, DataLayout> > gpu_t_right(d_t_right, 3, 31, 7, 20, 1); in test_cuda_contraction() local
302 gpu_t_result.device(gpu_device) = gpu_t_left.contract(gpu_t_right, dims); in test_cuda_contraction()
|