1#pragma once
2
3// @generated by torchgen/gen.py from Operator.h
4
5#include <tuple>
6#include <vector>
7
8// Forward declarations of any types needed in the operator signatures.
9// We can't directly include these classes because it will cause circular include dependencies.
10// This file is included by TensorBody.h, which defines the Tensor class.
11#include <ATen/core/ATen_fwd.h>
12
13namespace at {
14namespace _ops {
15
16
17struct TORCH_API _foreach_add_Scalar {
18 using schema = ::std::vector<at::Tensor> (at::TensorList, const at::Scalar &);
19 using ptr_schema = schema*;
20 // See Note [static constexpr char* members for windows NVCC]
21 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(name, "aten::_foreach_add")
22 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(overload_name, "Scalar")
23 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(schema_str, "_foreach_add.Scalar(Tensor[] self, Scalar scalar) -> Tensor[]")
24 static ::std::vector<at::Tensor> call(at::TensorList self, const at::Scalar & scalar);
25 static ::std::vector<at::Tensor> redispatch(c10::DispatchKeySet dispatchKeySet, at::TensorList self, const at::Scalar & scalar);
26};
27
28struct TORCH_API _foreach_add__Scalar {
29 using schema = void (at::TensorList, const at::Scalar &);
30 using ptr_schema = schema*;
31 // See Note [static constexpr char* members for windows NVCC]
32 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(name, "aten::_foreach_add_")
33 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(overload_name, "Scalar")
34 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(schema_str, "_foreach_add_.Scalar(Tensor(a!)[] self, Scalar scalar) -> ()")
35 static void call(at::TensorList self, const at::Scalar & scalar);
36 static void redispatch(c10::DispatchKeySet dispatchKeySet, at::TensorList self, const at::Scalar & scalar);
37};
38
39struct TORCH_API _foreach_add_List {
40 using schema = ::std::vector<at::Tensor> (at::TensorList, at::TensorList, const at::Scalar &);
41 using ptr_schema = schema*;
42 // See Note [static constexpr char* members for windows NVCC]
43 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(name, "aten::_foreach_add")
44 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(overload_name, "List")
45 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(schema_str, "_foreach_add.List(Tensor[] self, Tensor[] other, *, Scalar alpha=1) -> Tensor[]")
46 static ::std::vector<at::Tensor> call(at::TensorList self, at::TensorList other, const at::Scalar & alpha);
47 static ::std::vector<at::Tensor> redispatch(c10::DispatchKeySet dispatchKeySet, at::TensorList self, at::TensorList other, const at::Scalar & alpha);
48};
49
50struct TORCH_API _foreach_add__List {
51 using schema = void (at::TensorList, at::TensorList, const at::Scalar &);
52 using ptr_schema = schema*;
53 // See Note [static constexpr char* members for windows NVCC]
54 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(name, "aten::_foreach_add_")
55 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(overload_name, "List")
56 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(schema_str, "_foreach_add_.List(Tensor(a!)[] self, Tensor[] other, *, Scalar alpha=1) -> ()")
57 static void call(at::TensorList self, at::TensorList other, const at::Scalar & alpha);
58 static void redispatch(c10::DispatchKeySet dispatchKeySet, at::TensorList self, at::TensorList other, const at::Scalar & alpha);
59};
60
61struct TORCH_API _foreach_add_ScalarList {
62 using schema = ::std::vector<at::Tensor> (at::TensorList, at::ArrayRef<at::Scalar>);
63 using ptr_schema = schema*;
64 // See Note [static constexpr char* members for windows NVCC]
65 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(name, "aten::_foreach_add")
66 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(overload_name, "ScalarList")
67 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(schema_str, "_foreach_add.ScalarList(Tensor[] self, Scalar[] scalars) -> Tensor[]")
68 static ::std::vector<at::Tensor> call(at::TensorList self, at::ArrayRef<at::Scalar> scalars);
69 static ::std::vector<at::Tensor> redispatch(c10::DispatchKeySet dispatchKeySet, at::TensorList self, at::ArrayRef<at::Scalar> scalars);
70};
71
72struct TORCH_API _foreach_add__ScalarList {
73 using schema = void (at::TensorList, at::ArrayRef<at::Scalar>);
74 using ptr_schema = schema*;
75 // See Note [static constexpr char* members for windows NVCC]
76 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(name, "aten::_foreach_add_")
77 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(overload_name, "ScalarList")
78 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(schema_str, "_foreach_add_.ScalarList(Tensor(a!)[] self, Scalar[] scalars) -> ()")
79 static void call(at::TensorList self, at::ArrayRef<at::Scalar> scalars);
80 static void redispatch(c10::DispatchKeySet dispatchKeySet, at::TensorList self, at::ArrayRef<at::Scalar> scalars);
81};
82
83struct TORCH_API _foreach_add_Scalar_out {
84 using schema = void (at::TensorList, const at::Scalar &, at::TensorList);
85 using ptr_schema = schema*;
86 // See Note [static constexpr char* members for windows NVCC]
87 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(name, "aten::_foreach_add")
88 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(overload_name, "Scalar_out")
89 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(schema_str, "_foreach_add.Scalar_out(Tensor[] self, Scalar scalar, *, Tensor(a!)[] out) -> ()")
90 static void call(at::TensorList self, const at::Scalar & scalar, at::TensorList out);
91 static void redispatch(c10::DispatchKeySet dispatchKeySet, at::TensorList self, const at::Scalar & scalar, at::TensorList out);
92};
93
94struct TORCH_API _foreach_add_List_out {
95 using schema = void (at::TensorList, at::TensorList, const at::Scalar &, at::TensorList);
96 using ptr_schema = schema*;
97 // See Note [static constexpr char* members for windows NVCC]
98 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(name, "aten::_foreach_add")
99 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(overload_name, "List_out")
100 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(schema_str, "_foreach_add.List_out(Tensor[] self, Tensor[] other, *, Scalar alpha=1, Tensor(a!)[] out) -> ()")
101 static void call(at::TensorList self, at::TensorList other, const at::Scalar & alpha, at::TensorList out);
102 static void redispatch(c10::DispatchKeySet dispatchKeySet, at::TensorList self, at::TensorList other, const at::Scalar & alpha, at::TensorList out);
103};
104
105struct TORCH_API _foreach_add_ScalarList_out {
106 using schema = void (at::TensorList, at::ArrayRef<at::Scalar>, at::TensorList);
107 using ptr_schema = schema*;
108 // See Note [static constexpr char* members for windows NVCC]
109 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(name, "aten::_foreach_add")
110 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(overload_name, "ScalarList_out")
111 STATIC_CONSTEXPR_STR_INL_EXCEPT_WIN_CUDA(schema_str, "_foreach_add.ScalarList_out(Tensor[] self, Scalar[] scalars, *, Tensor(a!)[] out) -> ()")
112 static void call(at::TensorList self, at::ArrayRef<at::Scalar> scalars, at::TensorList out);
113 static void redispatch(c10::DispatchKeySet dispatchKeySet, at::TensorList self, at::ArrayRef<at::Scalar> scalars, at::TensorList out);
114};
115
116}} // namespace at::_ops
117