| 303 | } |
| 304 | |
| 305 | bool InferenceContext::FullyDefined(ShapeHandle s) { |
| 306 | if (!RankKnown(s)) return false; |
| 307 | for (int i = 0; i < Rank(s); ++i) { |
| 308 | if (!ValueKnown(Dim(s, i))) return false; |
| 309 | } |
| 310 | return true; |
| 311 | } |
| 312 | |
| 313 | DimensionHandle InferenceContext::NumElements(ShapeHandle s) { |
| 314 | const auto rank = Rank(s); |