Searched refs:TensorRankConstraint (Results 1 – 1 of 1) sorted by relevance
142 class TensorRankConstraint { class144 TensorRankConstraint(const TensorRankConstraint& copyFrom) { in TensorRankConstraint() function in __anon37b312a30111::TensorRankConstraint150 TensorRankConstraint& operator=(const TensorRankConstraint& copyFrom) { in operator =()157 static TensorRankConstraint Exactly(uint32_t rank) { in Exactly()158 return TensorRankConstraint(std::optional(rank), std::optional(rank)); in Exactly()161 static TensorRankConstraint AtLeast(uint32_t min) { in AtLeast()162 return TensorRankConstraint(std::optional(min), std::nullopt); in AtLeast()165 static TensorRankConstraint UpTo(uint32_t max) { in UpTo()166 return TensorRankConstraint(std::nullopt, std::optional(max)); in UpTo()169 static TensorRankConstraint Between(uint32_t min, uint32_t max) { in Between()[all …]