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/cat_ops.h>
21 :
22 : namespace at {
23 :
24 :
25 : // aten::cat(Tensor[] tensors, int dim=0) -> Tensor
26 378 : inline at::Tensor cat(const at::ITensorListRef & tensors, int64_t dim=0) {
27 378 : return at::_ops::cat::call(tensors, dim);
28 : }
29 :
30 : // aten::cat.out(Tensor[] tensors, int dim=0, *, Tensor(a!) out) -> Tensor(a!)
31 : inline at::Tensor & cat_out(at::Tensor & out, const at::ITensorListRef & tensors, int64_t dim=0) {
32 : return at::_ops::cat_out::call(tensors, dim, out);
33 : }
34 : // aten::cat.out(Tensor[] tensors, int dim=0, *, Tensor(a!) out) -> Tensor(a!)
35 : inline at::Tensor & cat_outf(const at::ITensorListRef & tensors, int64_t dim, at::Tensor & out) {
36 : return at::_ops::cat_out::call(tensors, dim, out);
37 : }
38 :
39 : // aten::cat.names(Tensor[] tensors, Dimname dim) -> Tensor
40 : inline at::Tensor cat(at::TensorList tensors, at::Dimname dim) {
41 : return at::_ops::cat_names::call(tensors, dim);
42 : }
43 :
44 : // aten::cat.names_out(Tensor[] tensors, Dimname dim, *, Tensor(a!) out) -> Tensor(a!)
45 : inline at::Tensor & cat_out(at::Tensor & out, at::TensorList tensors, at::Dimname dim) {
46 : return at::_ops::cat_names_out::call(tensors, dim, out);
47 : }
48 : // aten::cat.names_out(Tensor[] tensors, Dimname dim, *, Tensor(a!) out) -> Tensor(a!)
49 : inline at::Tensor & cat_outf(at::TensorList tensors, at::Dimname dim, at::Tensor & out) {
50 : return at::_ops::cat_names_out::call(tensors, dim, out);
51 : }
52 :
53 : }
|