Searched refs:max_kernel_impl (Results 1 – 2 of 2) sorted by relevance
57 void max_kernel_impl(const Tensor& result, const Tensor& indice, const Tensor& self, int64_t dim, b… in max_kernel_impl() function94 REGISTER_CUDA_DISPATCH(max_stub, &max_kernel_impl);
134 static void max_kernel_impl( in max_kernel_impl() function403 REGISTER_DISPATCH(max_stub, &max_kernel_impl);