| namespace at { | |
| struct TensorIterator; | |
| } | |
| namespace c10 { | |
| class Scalar; | |
| } | |
| namespace at { namespace native { | |
| void norm_launch_kernel(TensorIterator &iter, double val); | |
| void min_launch_kernel(TensorIterator &iter); | |
| void max_launch_kernel(TensorIterator &iter); | |
| void aminmax_launch_kernel(TensorIterator &iter); | |
| void min_all_launch_kernel(TensorIterator &iter); | |
| void max_all_launch_kernel(TensorIterator &iter); | |
| void aminmax_allreduce_launch_kernel(TensorIterator &iter); | |
| }} // namespace at::native |