Searched defs:redux_gpu (Results 1 – 2 of 2) sorted by relevance
86 Tensor<Type, 2, DataLayout> redux_gpu(dim_y, dim_z); in test_first_dim_reductions() local126 Tensor<Type, 2, DataLayout> redux_gpu(dim_x, dim_y); in test_last_dim_reductions() local
68 Tensor<float, 2> redux_gpu(reduced_tensorRange); in test_first_dim_reductions_sycl() local106 Tensor<float, 2> redux_gpu(reduced_tensorRange); in test_last_dim_reductions_sycl() local