Searched defs:gpu_in (Results 1 – 4 of 4) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_argmax_cuda.cu | 49 …Eigen::TensorMap<Eigen::Tensor<double, 3, Layout>, Aligned > gpu_in(d_in, Eigen::array<DenseIndex,… in test_cuda_simple_argmax() local
111 …Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout>, Aligned > gpu_in(d_in, Eigen::array<DenseInd… in test_cuda_argmax_dim() local
200 …Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout>, Aligned > gpu_in(d_in, Eigen::array<DenseInd… in test_cuda_argmin_dim() local
|
| D | cxx11_tensor_reduction_cuda.cu | 79 Eigen::TensorMap<Eigen::Tensor<Type, 3, DataLayout> > gpu_in(in_data, dim_x, dim_y, dim_z); in test_first_dim_reductions() local
119 Eigen::TensorMap<Eigen::Tensor<Type, 3, DataLayout> > gpu_in(in_data, dim_x, dim_y, dim_z); in test_last_dim_reductions() local
|
| D | cxx11_tensor_broadcast_sycl.cpp | 50 TensorMap<Tensor<float, 4>> gpu_in(gpu_in_data, in_range); in test_broadcast_sycl() local
|
| D | cxx11_tensor_cuda.cu | 642 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_in(d_in, 72, 97); in test_cuda_lgamma() local
696 Eigen::TensorMap<Eigen::Tensor<Scalar, 1> > gpu_in(d_in, 7); in test_cuda_digamma() local
1011 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_in(d_in, 72, 97); in test_cuda_erf() local
1050 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_in(d_in, 72, 97); in test_cuda_erfc() local
|