Searched refs:max_all_kernel_impl (Results 1 – 2 of 2) sorted by relevance
77 void max_all_kernel_impl(Tensor& result, const Tensor& input) { in max_all_kernel_impl() function96 REGISTER_CUDA_DISPATCH(max_all_stub, &max_all_kernel_impl);
90 static void max_all_kernel_impl(Tensor& result, const Tensor& input) { in max_all_kernel_impl() function224 REGISTER_DISPATCH(max_all_stub, &max_all_kernel_impl);