Searched defs:gpu_input (Results 1 – 1 of 1) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_cuda.cu | 344 Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout> > gpu_input(d_input, 74,37,11,137); in test_cuda_convolution_1d() local
397 Eigen::TensorMap<Eigen::Tensor<float, 4, ColMajor> > gpu_input(d_input,74,9,11,7); in test_cuda_convolution_inner_dim_col_major_1d() local
450 Eigen::TensorMap<Eigen::Tensor<float, 4, RowMajor> > gpu_input(d_input, 7,9,11,74); in test_cuda_convolution_inner_dim_row_major_1d() local
504 Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout> > gpu_input(d_input,74,37,11,137); in test_cuda_convolution_2d() local
568 Eigen::TensorMap<Eigen::Tensor<float, 5, DataLayout> > gpu_input(d_input,74,37,11,137,17); in test_cuda_convolution_3d() local
|