Searched refs:dim_z (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_reduction_cuda.cu | 63 int dim_z = 128; in test_first_dim_reductions() local
65 Tensor<Type, 3, DataLayout> in(dim_x, dim_y, dim_z); in test_first_dim_reductions()
77 Type* in_data = (Type*)dev.allocate(dim_x*dim_y*dim_z*sizeof(Type)); in test_first_dim_reductions()
78 Type* out_data = (Type*)dev.allocate(dim_z*dim_y*sizeof(Type)); in test_first_dim_reductions()
79 Eigen::TensorMap<Eigen::Tensor<Type, 3, DataLayout> > gpu_in(in_data, dim_x, dim_y, dim_z); in test_first_dim_reductions()
80 Eigen::TensorMap<Eigen::Tensor<Type, 2, DataLayout> > gpu_out(out_data, dim_y, dim_z); in test_first_dim_reductions()
86 Tensor<Type, 2, DataLayout> redux_gpu(dim_y, dim_z); in test_first_dim_reductions()
103 int dim_z = 33; in test_last_dim_reductions() local
105 Tensor<Type, 3, DataLayout> in(dim_x, dim_y, dim_z); in test_last_dim_reductions()
117 Type* in_data = (Type*)dev.allocate(dim_x*dim_y*dim_z*sizeof(Type)); in test_last_dim_reductions()
[all …]
|
| D | cxx11_tensor_reduction_sycl.cpp | 59 int dim_z = 67; in test_first_dim_reductions_sycl() local
61 array<int, 3> tensorRange = {{dim_x, dim_y, dim_z}}; in test_first_dim_reductions_sycl()
64 array<int, 2> reduced_tensorRange = {{dim_y, dim_z}}; in test_first_dim_reductions_sycl()
97 int dim_z = 47; in test_last_dim_reductions_sycl() local
99 array<int, 3> tensorRange = {{dim_x, dim_y, dim_z}}; in test_last_dim_reductions_sycl()
|