Searched refs:reduced_tensorRange (Results 1 – 1 of 1) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_reduction_sycl.cpp | 64 array<int, 2> reduced_tensorRange = {{dim_y, dim_z}}; in test_first_dim_reductions_sycl() local
67 Tensor<float, 2> redux(reduced_tensorRange); in test_first_dim_reductions_sycl()
68 Tensor<float, 2> redux_gpu(reduced_tensorRange); in test_first_dim_reductions_sycl()
78 TensorMap<Tensor<float, 2> > out_gpu(gpu_out_data, reduced_tensorRange); in test_first_dim_reductions_sycl()
85 for(int j=0; j<reduced_tensorRange[0]; j++ ) in test_first_dim_reductions_sycl()
86 for(int k=0; k<reduced_tensorRange[1]; k++ ) in test_first_dim_reductions_sycl()
102 array<int, 2> reduced_tensorRange = {{dim_x, dim_y}}; in test_last_dim_reductions_sycl() local
105 Tensor<float, 2> redux(reduced_tensorRange); in test_last_dim_reductions_sycl()
106 Tensor<float, 2> redux_gpu(reduced_tensorRange); in test_last_dim_reductions_sycl()
116 TensorMap<Tensor<float, 2> > out_gpu(gpu_out_data, reduced_tensorRange); in test_last_dim_reductions_sycl()
[all …]
|