ReduceAllOps.h 387 B

12345678910111213141516
  1. #pragma once
  2. #include <ATen/native/DispatchStub.h>
  3. namespace at {
  4. class Tensor;
  5. }
  6. namespace at { namespace native {
  7. using reduce_all_fn = void (*)(Tensor & result, const Tensor & self);
  8. using reduce_min_max_fn = void (*)(Tensor & max_result, Tensor & min_result, const Tensor & self);
  9. DECLARE_DISPATCH(reduce_all_fn, min_all_stub);
  10. DECLARE_DISPATCH(reduce_all_fn, max_all_stub);
  11. }}