| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_argmax_gpu.cu | 48 …Eigen::TensorMap<Eigen::Tensor<double, 3, Layout>, Aligned > gpu_in(d_in, Eigen::array<DenseIndex,… in test_gpu_simple_argmax() local
110 …Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout>, Aligned > gpu_in(d_in, Eigen::array<DenseInd… in test_gpu_argmax_dim() local
199 …Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout>, Aligned > gpu_in(d_in, Eigen::array<DenseInd… in test_gpu_argmin_dim() local
|
| H A D | cxx11_tensor_argmax_sycl.cpp | 47 Eigen::TensorMap<Eigen::Tensor<DataType, 3, Layout, DenseIndex> > gpu_in(d_in, in test_sycl_simple_argmax() local
109 Eigen::TensorMap<Eigen::Tensor<DataType, 4, DataLayout, DenseIndex> > gpu_in( in test_sycl_argmax_dim() local
196 Eigen::TensorMap<Eigen::Tensor<DataType, 4, DataLayout, DenseIndex> > gpu_in( in test_sycl_argmin_dim() local
|
| H A D | cxx11_tensor_gpu.cu | 644 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_in(d_in, 72, 97); in test_gpu_lgamma() local
699 Eigen::TensorMap<Eigen::Tensor<Scalar, 1> > gpu_in(d_in, 7); in test_gpu_digamma() local
1015 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_in(d_in, 72, 97); in test_gpu_erf() local
1054 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_in(d_in, 72, 97); in test_gpu_erfc() local
1317 Eigen::TensorMap<Eigen::Tensor<Scalar, 1> > gpu_in(d_in, 21); in test_gpu_i0e() local
1372 Eigen::TensorMap<Eigen::Tensor<Scalar, 1> > gpu_in(d_in, 21); in test_gpu_i1e() local
|
| H A D | cxx11_tensor_reduction_gpu.cu | 76 Eigen::TensorMap<Eigen::Tensor<Type, 3, DataLayout> > gpu_in(in_data, dim_x, dim_y, dim_z); in test_first_dim_reductions() local
116 Eigen::TensorMap<Eigen::Tensor<Type, 3, DataLayout> > gpu_in(in_data, dim_x, dim_y, dim_z); in test_last_dim_reductions() local
|
| H A D | cxx11_tensor_broadcast_sycl.cpp | 59 …TensorMap<TensorFixedSize<DataType, Sizes<2, 3, 5, 7>, DataLayout, IndexType>> gpu_in(gpu_in_data,… in test_broadcast_sycl_fixed() local
110 TensorMap<Tensor<DataType, 4, DataLayout, IndexType>> gpu_in(gpu_in_data, in_range); in test_broadcast_sycl() local
|
| H A D | cxx11_tensor_sycl.cpp | 328 TensorMap<Tensor<Scalar1, 1, DataLayout, IndexType>> gpu_in(gpu_in_data, tensorRange); in test_sycl_cast() local
|