Searched defs:gpu_input (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_convolution_sycl.cpp | 68 …Eigen::TensorMap<Eigen::Tensor<DataType, 3, DataLayout, IndexType> > gpu_input(d_input, input_dims… in test_larg_expr1D() local
129 …Eigen::TensorMap<Eigen::Tensor<DataType, 3, DataLayout, IndexType> > gpu_input(d_input, input_dims… in test_larg_expr2D() local
190 …Eigen::TensorMap<Eigen::Tensor<DataType, 3, DataLayout, IndexType> > gpu_input(d_input, input_dims… in test_larg_expr3D() local
243 …Eigen::TensorMap<Eigen::Tensor<DataType, 2, DataLayout, IndexType> > gpu_input(d_input, input_dims… in test_evals() local
289 …Eigen::TensorMap<Eigen::Tensor<DataType, 2, DataLayout,IndexType> > gpu_input(d_input, input_dims); in test_expr() local
349 …Eigen::TensorMap<Eigen::Tensor<DataType, 1, DataLayout,IndexType> > gpu_input(d_input, input_dims); in test_modes() local
430 …Eigen::TensorMap<Eigen::Tensor<DataType, 1, DataLayout,IndexType> > gpu_input(d_input, input_dims); in test_strides() local
|
| H A D | cxx11_tensor_gpu.cu | 345 Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout> > gpu_input(d_input, 74,37,11,137); in test_gpu_convolution_1d() local
398 Eigen::TensorMap<Eigen::Tensor<float, 4, ColMajor> > gpu_input(d_input,74,9,11,7); in test_gpu_convolution_inner_dim_col_major_1d() local
451 Eigen::TensorMap<Eigen::Tensor<float, 4, RowMajor> > gpu_input(d_input, 7,9,11,74); in test_gpu_convolution_inner_dim_row_major_1d() local
505 Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout> > gpu_input(d_input,74,37,11,137); in test_gpu_convolution_2d() local
569 Eigen::TensorMap<Eigen::Tensor<float, 5, DataLayout> > gpu_input(d_input,74,37,11,137,17); in test_gpu_convolution_3d() local
|