forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathReduceAllOps.cpp
35 lines (28 loc) · 1.2 KB
/
ReduceAllOps.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#include <ATen/native/ReduceAllOps.h>
#include <ATen/ATen.h>
#include <ATen/NativeFunctions.h>
namespace at {
namespace native {
DEFINE_DISPATCH(min_all_stub);
DEFINE_DISPATCH(max_all_stub);
Tensor min(const Tensor &self) {
TORCH_CHECK(self.numel() > 0,
"min(): Expected reduction dim to be specified for input.numel() == 0. Specify the reduction dim with the 'dim' argument.");
Tensor result = at::empty({}, self.options());
min_all_stub(self.device().type(), result, self.contiguous());
return result;
}
Tensor max(const Tensor &self) {
TORCH_CHECK(self.numel() > 0,
"max(): Expected reduction dim to be specified for input.numel() == 0. Specify the reduction dim with the 'dim' argument.");
Tensor result = at::empty({}, self.options());
max_all_stub(self.device().type(), result, self.contiguous());
return result;
}
// DEPRECATED: Use at::aminmax instead
std::tuple<Tensor, Tensor> _aminmax_all(const Tensor &self) {
TORCH_WARN_ONCE("_aminmax is deprecated as of PyTorch 1.11 and will be removed in a future release. Use aminmax instead."
" This warning will only appear once per process.");
return at::aminmax(self);
}
}} // namespace at::native