Searched refs:TensorGeometryArg (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | TensorUtils.h | 40 struct TORCH_API TensorGeometryArg { struct
44 /* implicit */ TensorGeometryArg(TensorArg arg) in TensorGeometryArg() argument
46 TensorGeometryArg(TensorGeometry tensor, const char* name, int pos) in TensorGeometryArg() function
73 const TensorGeometryArg& t);
80 TORCH_API void checkDim(CheckedFrom c, const TensorGeometryArg& t, int64_t dim);
84 const TensorGeometryArg& t,
89 const TensorGeometryArg& t1,
90 const TensorGeometryArg& t2);
91 TORCH_API void checkContiguous(CheckedFrom c, const TensorGeometryArg& t);
95 const TensorGeometryArg& t,
[all …]
|
| H A D | TensorUtils.cpp | 12 std::ostream& operator<<(std::ostream & out, const TensorGeometryArg& t) { in operator <<()
36 TensorGeometryArg(TensorArg({tensor, name, pos})), in checkDim()
42 void checkDim(CheckedFrom c, const TensorGeometryArg& t, int64_t dim) { in checkDim()
48 void checkDimRange(CheckedFrom c, const TensorGeometryArg& t, int64_t dim_start, int64_t dim_end) { in checkDimRange()
56 void checkContiguous(CheckedFrom c, const TensorGeometryArg& t) { in checkContiguous()
70 void checkSize(CheckedFrom c, const TensorGeometryArg& t, IntArrayRef sizes) { in checkSize()
78 void checkSize_symint(CheckedFrom c, const TensorGeometryArg& t, c10::SymIntArrayRef sizes) { in checkSize_symint()
86 void checkSize(CheckedFrom c, const TensorGeometryArg& t, int64_t dim, int64_t size) { in checkSize()
94 void checkSize_symint(CheckedFrom c, const TensorGeometryArg& t, int64_t dim, const c10::SymInt& si… in checkSize_symint()
126 void checkNumel(CheckedFrom c, const TensorGeometryArg& t, int64_t numel) { in checkNumel()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ConvUtils.h | 193 … const TensorGeometryArg& input, const TensorGeometryArg& weight, const TensorGeometryArg& output, in convolution_shape_check()
|