xref: /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ReduceOps.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 
2 namespace at {
3 struct TensorIterator;
4 }
5 
6 namespace c10 {
7 class Scalar;
8 }
9 
10 namespace at { namespace native {
11 
12 void norm_launch_kernel(TensorIterator &iter, double val);
13 void min_launch_kernel(TensorIterator &iter);
14 void max_launch_kernel(TensorIterator &iter);
15 void aminmax_launch_kernel(TensorIterator &iter);
16 void min_all_launch_kernel(TensorIterator &iter);
17 void max_all_launch_kernel(TensorIterator &iter);
18 void aminmax_allreduce_launch_kernel(TensorIterator &iter);
19 
20 }}  // namespace at::native
21