Searched refs: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
349 gpu_out.device(gpu_device) = gpu_input.convolve(gpu_kernel, dims); in test_cuda_convolution_1d()
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
402 gpu_out.device(gpu_device) = gpu_input.convolve(gpu_kernel, dims); in test_cuda_convolution_inner_dim_col_major_1d()
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
455 gpu_out.device(gpu_device) = gpu_input.convolve(gpu_kernel, dims); in test_cuda_convolution_inner_dim_row_major_1d()
504 Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout> > gpu_input(d_input,74,37,11,137); in test_cuda_convolution_2d() local
509 gpu_out.device(gpu_device) = gpu_input.convolve(gpu_kernel, dims); in test_cuda_convolution_2d()
568 Eigen::TensorMap<Eigen::Tensor<float, 5, DataLayout> > gpu_input(d_input,74,37,11,137,17); in test_cuda_convolution_3d() local
573 gpu_out.device(gpu_device) = gpu_input.convolve(gpu_kernel, dims); in test_cuda_convolution_3d()
|