Home
last modified time | relevance | path

Searched refs:gpu_in (Results 1 – 4 of 4) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_argmax_cuda.cu49 …Eigen::TensorMap<Eigen::Tensor<double, 3, Layout>, Aligned > gpu_in(d_in, Eigen::array<DenseIndex,… in test_cuda_simple_argmax() local
53 gpu_out_max.device(gpu_device) = gpu_in.argmax(); in test_cuda_simple_argmax()
54 gpu_out_min.device(gpu_device) = gpu_in.argmin(); in test_cuda_simple_argmax()
111 …Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout>, Aligned > gpu_in(d_in, Eigen::array<DenseInd… in test_cuda_argmax_dim() local
114 gpu_out.device(gpu_device) = gpu_in.argmax(dim); in test_cuda_argmax_dim()
142 gpu_out.device(gpu_device) = gpu_in.argmax(dim); in test_cuda_argmax_dim()
200 …Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout>, Aligned > gpu_in(d_in, Eigen::array<DenseInd… in test_cuda_argmin_dim() local
203 gpu_out.device(gpu_device) = gpu_in.argmin(dim); in test_cuda_argmin_dim()
231 gpu_out.device(gpu_device) = gpu_in.argmin(dim); in test_cuda_argmin_dim()
Dcxx11_tensor_reduction_cuda.cu79 Eigen::TensorMap<Eigen::Tensor<Type, 3, DataLayout> > gpu_in(in_data, dim_x, dim_y, dim_z); in test_first_dim_reductions() local
84 gpu_out.device(dev) = gpu_in.sum(red_axis); in test_first_dim_reductions()
85 gpu_out.device(dev) += gpu_in.sum(red_axis); in test_first_dim_reductions()
119 Eigen::TensorMap<Eigen::Tensor<Type, 3, DataLayout> > gpu_in(in_data, dim_x, dim_y, dim_z); in test_last_dim_reductions() local
124 gpu_out.device(dev) = gpu_in.sum(red_axis); in test_last_dim_reductions()
125 gpu_out.device(dev) += gpu_in.sum(red_axis); in test_last_dim_reductions()
Dcxx11_tensor_broadcast_sycl.cpp50 TensorMap<Tensor<float, 4>> gpu_in(gpu_in_data, in_range); in test_broadcast_sycl() local
53 gpu_out.device(sycl_device) = gpu_in.broadcast(broadcasts); in test_broadcast_sycl()
Dcxx11_tensor_cuda.cu642 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_in(d_in, 72, 97); in test_cuda_lgamma() local
645 gpu_out.device(gpu_device) = gpu_in.lgamma(); in test_cuda_lgamma()
696 Eigen::TensorMap<Eigen::Tensor<Scalar, 1> > gpu_in(d_in, 7); in test_cuda_digamma() local
699 gpu_out.device(gpu_device) = gpu_in.digamma(); in test_cuda_digamma()
1011 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_in(d_in, 72, 97); in test_cuda_erf() local
1014 gpu_out.device(gpu_device) = gpu_in.erf(); in test_cuda_erf()
1050 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_in(d_in, 72, 97); in test_cuda_erfc() local
1053 gpu_out.device(gpu_device) = gpu_in.erfc(); in test_cuda_erfc()