Line data    Source code

       1              : #pragma once
       2              : 
       3              : // @generated by torchgen/gen.py from Function.h
       4              : 
       5              : #include <ATen/Context.h>
       6              : #include <ATen/DeviceGuard.h>
       7              : #include <ATen/TensorUtils.h>
       8              : #include <ATen/TracerMode.h>
       9              : #include <ATen/core/Generator.h>
      10              : #include <ATen/core/Reduction.h>
      11              : #include <ATen/core/Tensor.h>
      12              : #include <c10/core/Scalar.h>
      13              : #include <c10/core/Storage.h>
      14              : #include <c10/core/TensorOptions.h>
      15              : #include <c10/util/Deprecated.h>
      16              : #include <c10/util/Optional.h>
      17              : 
      18              : 
      19              : 
      20              : #include <ATen/ops/argmax_ops.h>
      21              : 
      22              : namespace at {
      23              : 
      24              : 
      25              : // aten::argmax(Tensor self, int? dim=None, bool keepdim=False) -> Tensor
      26          542 : inline at::Tensor argmax(const at::Tensor & self, c10::optional<int64_t> dim=c10::nullopt, bool keepdim=false) {
      27          542 :     return at::_ops::argmax::call(self, dim, keepdim);
      28              : }
      29              : 
      30              : // aten::argmax.out(Tensor self, int? dim=None, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!)
      31              : inline at::Tensor & argmax_out(at::Tensor & out, const at::Tensor & self, c10::optional<int64_t> dim=c10::nullopt, bool keepdim=false) {
      32              :     return at::_ops::argmax_out::call(self, dim, keepdim, out);
      33              : }
      34              : // aten::argmax.out(Tensor self, int? dim=None, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!)
      35              : inline at::Tensor & argmax_outf(const at::Tensor & self, c10::optional<int64_t> dim, bool keepdim, at::Tensor & out) {
      36              :     return at::_ops::argmax_out::call(self, dim, keepdim, out);
      37              : }
      38              : 
      39              : }