ReduceOps.h 500 B

1234567891011121314151617181920
  1. namespace at {
  2. struct TensorIterator;
  3. }
  4. namespace c10 {
  5. class Scalar;
  6. }
  7. namespace at { namespace native {
  8. void norm_launch_kernel(TensorIterator &iter, double val);
  9. void min_launch_kernel(TensorIterator &iter);
  10. void max_launch_kernel(TensorIterator &iter);
  11. void aminmax_launch_kernel(TensorIterator &iter);
  12. void min_all_launch_kernel(TensorIterator &iter);
  13. void max_all_launch_kernel(TensorIterator &iter);
  14. void aminmax_allreduce_launch_kernel(TensorIterator &iter);
  15. }} // namespace at::native