Lines Matching defs:gpu_out

93   Eigen::TensorMap<Eigen::Tensor<float, 1>, Eigen::Aligned> gpu_out(  in test_cuda_elementwise_small()  local
147 …Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_out(d_out, Eigen::array<Eigen::DenseIndex, 3>(72,53… in test_cuda_elementwise() local
188 Eigen::TensorMap<Eigen::Tensor<bool, 1>, Eigen::Aligned> gpu_out( in test_cuda_props() local
225 Eigen::TensorMap<Eigen::Tensor<float, 2> > gpu_out(d_out, 72,97); in test_cuda_reduction() local
343 Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout> > gpu_out(d_out, 74,34,11,137); in test_cuda_convolution_1d() local
396 Eigen::TensorMap<Eigen::Tensor<float, 4, ColMajor> > gpu_out(d_out,71,9,11,7); in test_cuda_convolution_inner_dim_col_major_1d() local
449 Eigen::TensorMap<Eigen::Tensor<float, 4, RowMajor> > gpu_out(d_out, 7,9,11,71); in test_cuda_convolution_inner_dim_row_major_1d() local
503 Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout> > gpu_out(d_out,74,35,8,137); in test_cuda_convolution_2d() local
567 Eigen::TensorMap<Eigen::Tensor<float, 5, DataLayout> > gpu_out(d_out,74,35,8,136,17); in test_cuda_convolution_3d() local
640 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_out(d_out, 72, 97); in test_cuda_lgamma() local
694 Eigen::TensorMap<Eigen::Tensor<Scalar, 1> > gpu_out(d_out, 7); in test_cuda_digamma() local
759 Eigen::TensorMap<Eigen::Tensor<Scalar, 1> > gpu_out(d_out, 6); in test_cuda_zeta() local
830 Eigen::TensorMap<Eigen::Tensor<Scalar, 1> > gpu_out(d_out, 7); in test_cuda_polygamma() local
895 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_out(d_out, 6, 6); in test_cuda_igamma() local
965 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_out(d_out, 6, 6); in test_cuda_igammac() local
1009 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_out(d_out, 72, 97); in test_cuda_erf() local
1048 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_out(d_out, 72, 97); in test_cuda_erfc() local
1190 Eigen::TensorMap<Eigen::Tensor<Scalar, 1> > gpu_out(d_out, 125); in test_cuda_betainc() local