Searched defs:VaryingShape (Results 1 – 2 of 2) sorted by relevance
500 struct VaryingShape { struct502 VaryingShape(const std::vector<T>& vec) in VaryingShape() argument505 VaryingShape(c10::ArrayRef<T> vec) in VaryingShape() function508 VaryingShape(std::optional<size_t> size = std::nullopt) : dims_(std::nullopt) { in dims_() argument514 VaryingShape(ListOfOptionalElements dims) : dims_(std::move(dims)) {} in VaryingShape() argument516 VaryingShape(size_t size) : VaryingShape(std::optional<size_t>(size)) {} in VaryingShape() argument541 TORCH_API VaryingShape merge(const VaryingShape& other) const; argument543 std::optional<std::vector<T>> concrete_sizes() const { in concrete_sizes()558 bool isComplete() const { in isComplete()
337 template struct VaryingShape<c10::ShapeSymbol>; variable338 template struct VaryingShape<bool>; variable339 template struct VaryingShape<size_t>; variable340 template struct VaryingShape<int64_t>; variable341 template struct VaryingShape<c10::Stride>; variable