#pragma once // @generated by torchgen/gen.py from Function.h #include #include #include #include #include #include #include #include #include #include #include #include #include namespace at { // aten::unflatten.int(Tensor(a) self, int dim, SymInt[] sizes) -> Tensor(a) inline at::Tensor unflatten(const at::Tensor & self, int64_t dim, at::IntArrayRef sizes) { return at::_ops::unflatten_int::call(self, dim, c10::fromIntArrayRefSlow(sizes)); } namespace symint { template ::value>> at::Tensor unflatten(const at::Tensor & self, int64_t dim, at::IntArrayRef sizes) { return at::_ops::unflatten_int::call(self, dim, c10::fromIntArrayRefSlow(sizes)); } } // aten::unflatten.int(Tensor(a) self, int dim, SymInt[] sizes) -> Tensor(a) inline at::Tensor unflatten_symint(const at::Tensor & self, int64_t dim, c10::SymIntArrayRef sizes) { return at::_ops::unflatten_int::call(self, dim, sizes); } namespace symint { template ::value>> at::Tensor unflatten(const at::Tensor & self, int64_t dim, c10::SymIntArrayRef sizes) { return at::_ops::unflatten_int::call(self, dim, sizes); } } // aten::unflatten.Dimname(Tensor(a) self, Dimname dim, SymInt[] sizes, Dimname[] names) -> Tensor(a) inline at::Tensor unflatten(const at::Tensor & self, at::Dimname dim, at::IntArrayRef sizes, at::DimnameList names) { return at::_ops::unflatten_Dimname::call(self, dim, c10::fromIntArrayRefSlow(sizes), names); } namespace symint { template ::value>> at::Tensor unflatten(const at::Tensor & self, at::Dimname dim, at::IntArrayRef sizes, at::DimnameList names) { return at::_ops::unflatten_Dimname::call(self, dim, c10::fromIntArrayRefSlow(sizes), names); } } // aten::unflatten.Dimname(Tensor(a) self, Dimname dim, SymInt[] sizes, Dimname[] names) -> Tensor(a) inline at::Tensor unflatten_symint(const at::Tensor & self, at::Dimname dim, c10::SymIntArrayRef sizes, at::DimnameList names) { return at::_ops::unflatten_Dimname::call(self, dim, sizes, names); } namespace symint { template ::value>> at::Tensor unflatten(const at::Tensor & self, at::Dimname dim, c10::SymIntArrayRef sizes, at::DimnameList names) { return at::_ops::unflatten_Dimname::call(self, dim, sizes, names); } } }