| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_cuda.cu | 96 Eigen::TensorMap<Eigen::Tensor<float, 1>, Eigen::Aligned> gpu_out( in test_cuda_elementwise_small() local
150 …Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_out(d_out, Eigen::array<Eigen::DenseIndex, 3>(72,53… in test_cuda_elementwise() local
191 Eigen::TensorMap<Eigen::Tensor<bool, 1>, Eigen::Aligned> gpu_out( in test_cuda_props() local
228 Eigen::TensorMap<Eigen::Tensor<float, 2> > gpu_out(d_out, 72,97); in test_cuda_reduction() local
346 Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout> > gpu_out(d_out, 74,34,11,137); in test_cuda_convolution_1d() local
399 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
452 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
506 Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout> > gpu_out(d_out,74,35,8,137); in test_cuda_convolution_2d() local
570 Eigen::TensorMap<Eigen::Tensor<float, 5, DataLayout> > gpu_out(d_out,74,35,8,136,17); in test_cuda_convolution_3d() local
643 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_out(d_out, 72, 97); in test_cuda_lgamma() local
[all …]
|
| D | cxx11_tensor_random_cuda.cu | 36 Eigen::TensorMap<Eigen::Tensor<float, 2> > gpu_out(d_out, 72,97); in test_cuda_random_uniform() local
61 Eigen::TensorMap<Eigen::Tensor<float, 2> > gpu_out(d_out, 72,97); in test_cuda_random_normal() local
|
| D | cxx11_tensor_reduction_cuda.cu | 80 Eigen::TensorMap<Eigen::Tensor<Type, 2, DataLayout> > gpu_out(out_data, dim_y, dim_z); in test_first_dim_reductions() local
120 Eigen::TensorMap<Eigen::Tensor<Type, 2, DataLayout> > gpu_out(out_data, dim_x, dim_y); in test_last_dim_reductions() local
|
| D | cxx11_tensor_argmax_cuda.cu | 112 Eigen::TensorMap<Eigen::Tensor<DenseIndex, 3, DataLayout>, Aligned > gpu_out(d_out, out_shape); in test_cuda_argmax_dim() local
201 Eigen::TensorMap<Eigen::Tensor<DenseIndex, 3, DataLayout>, Aligned > gpu_out(d_out, out_shape); in test_cuda_argmin_dim() local
|
| D | cxx11_tensor_complex_cwise_ops_cuda.cu | 41 Eigen::TensorMap<Eigen::Tensor<std::complex<T>, 1, 0, int>, Eigen::Aligned> gpu_out( in test_cuda_complex_cwise_ops() local
|
| D | cxx11_tensor_broadcast_sycl.cpp | 51 TensorMap<Tensor<float, 4>> gpu_out(gpu_out_data, out_range); in test_broadcast_sycl() local
|
| D | cxx11_tensor_forced_eval_sycl.cpp | 45 Eigen::TensorMap<Eigen::Tensor<float, 3>> gpu_out(gpu_out_data, tensorRange); in test_forced_eval_sycl() local
|
| D | cxx11_tensor_sycl.cpp | 52 TensorMap<Tensor<float, 3>> gpu_out(gpu_out_data, tensorRange); in test_sycl_cpu() local
|
| D | cxx11_tensor_device.cu | 295 Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_out(d_out, 40,50,70); in test_gpu() local
|