Searched defs:TensorGeometry (Results 1 – 1 of 1) sorted by relevance
15 struct TORCH_API TensorGeometry { struct18 explicit TensorGeometry(c10::SymIntArrayRef sizes) in TensorGeometry() argument32 explicit TensorGeometry(const TensorBase& t) in TensorGeometry() function43 int64_t dim() const { in dim()47 int64_t size(int64_t dim) const { in size()52 c10::IntArrayRef sizes() const { in sizes()56 int64_t stride(int64_t dim) const { in stride()61 c10::IntArrayRef strides() const { in strides()65 int64_t storage_offset() const { in storage_offset()69 int64_t numel() const { in numel()[all …]