7.4 KiB
7.4 KiB
<html lang="en">
<head>
</head>
</html>
LCOV - code coverage report | ||||||||||||||||||||||
![]() | ||||||||||||||||||||||
|
||||||||||||||||||||||
![]() |
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 : } |
![]() |
Generated by: LCOV version 2.0-1 |
</html>