Searched defs:redux_gpu (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_reduction_gpu.cu | 83 Tensor<Type, 2, DataLayout> redux_gpu(dim_y, dim_z); in test_first_dim_reductions() local
123 Tensor<Type, 2, DataLayout> redux_gpu(dim_x, dim_y); in test_last_dim_reductions() local
|
| D | cxx11_tensor_reduction_sycl.cpp | 454 Tensor<DataType, 2, DataLayout, IndexType> redux_gpu(reduced_tensorRange); in test_first_dim_reductions_max_sycl() local
501 reduced_tensor redux_gpu(reduced_range); in test_first_dim_reductions_max_with_offset_sycl() local
559 reduced_tensor redux_gpu(reduced_range); in test_last_dim_reductions_max_with_offset_sycl() local
618 Tensor<DataType, 1, DataLayout, IndexType> redux_gpu(reduced_tensorRange); in test_first_dim_reductions_sum_sycl() local
662 Tensor<DataType, 2, DataLayout, IndexType> redux_gpu(reduced_tensorRange); in test_first_dim_reductions_mean_sycl() local
708 Tensor<DataType, 2, DataLayout, IndexType> redux_gpu(reduced_tensorRange); in test_last_dim_reductions_mean_sycl() local
753 Tensor<DataType, 2, DataLayout, IndexType> redux_gpu(reduced_tensorRange); in test_last_dim_reductions_sum_sycl() local
|