Searched refs:redux_dim (Results 1 – 1 of 1) sorted by relevance
345 Eigen::array<int, 1> redux_dim = {{redux}}; in test_cuda_reductions() local346 gpu_res_float.device(gpu_device) = gpu_float1.sum(redux_dim).cast<Eigen::half>(); in test_cuda_reductions()347 gpu_res_half.device(gpu_device) = gpu_float1.cast<Eigen::half>().sum(redux_dim); in test_cuda_reductions()