| #pragma once | |
| #include <ATen/native/DispatchStub.h> | |
| namespace at { | |
| class Tensor; | |
| } | |
| namespace at { namespace native { | |
| using reduce_all_fn = void (*)(Tensor & result, const Tensor & self); | |
| using reduce_min_max_fn = void (*)(Tensor & max_result, Tensor & min_result, const Tensor & self); | |
| DECLARE_DISPATCH(reduce_all_fn, min_all_stub); | |
| DECLARE_DISPATCH(reduce_all_fn, max_all_stub); | |
| }} |