Searched refs:aminmax_allreduce_kernel_impl (Results 1 – 1 of 1) sorted by relevance
83 void aminmax_allreduce_kernel_impl(const Tensor& input, Tensor& min_result, Tensor& max_result) { in aminmax_allreduce_kernel_impl() function97 REGISTER_CUDA_DISPATCH(aminmax_allreduce_stub, &aminmax_allreduce_kernel_impl);