Searched defs:SymbolicShape (Results 1 – 4 of 4) sorted by relevance
373 struct TORCH_API SymbolicShape { struct375 SymbolicShape() : dims_(std::nullopt) {} in SymbolicShape() argument378 SymbolicShape(std::optional<size_t> rank) : dims_(std::nullopt) { in SymbolicShape() function392 SymbolicShape(const std::vector<std::optional<int64_t>>& dims) { in SymbolicShape() function407 SymbolicShape(std::vector<ShapeSymbol> dims) : dims_(std::move(dims)) {} in SymbolicShape() argument409 SymbolicShape(c10::IntArrayRef dims) { in SymbolicShape() argument425 ShapeSymbol at(size_t i) const { in at()433 std::optional<size_t> rank() const { in rank()440 std::optional<std::vector<ShapeSymbol>> sizes() const { in sizes()444 std::optional<std::vector<bool>> symbolicDims() const { in symbolicDims()[all …]
47 using SymbolicShape = SymbolicShapesResolver::SymbolicShape; typedef
40 using SymbolicShape = SymbolicShapesResolver::SymbolicShape; typedef
35 using SymbolicShape = SymbolicShapesResolver::SymbolicShape; typedef