Searched refs:gpu_t_result (Results 1 – 3 of 3) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_contract_cuda.cu | 65 gpu_t_result(d_t_result, Eigen::array<int, 2>(m_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()
129 gpu_t_result(d_t_result); 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_scan_cuda.cu | 52 gpu_t_result(d_t_result, Eigen::array<int, 3>(m_size, k_size, n_size)); in test_cuda_cumsum()
54 gpu_t_result.device(gpu_device) = gpu_t_input.cumsum(1); in test_cuda_cumsum()
|
| D | cxx11_tensor_cuda.cu | 289 Eigen::TensorMap<Eigen::Tensor<float, 5, DataLayout> > gpu_t_result(d_t_result, 6, 50, 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()
|