| 751 | } |
| 752 | |
| 753 | ShapeHandle InferenceContext::UnknownShapeOfRank(int64 rank) { |
| 754 | CHECK_LE(rank, kint32max) << "rank must be less than kint32max"; |
| 755 | if (rank == kUnknownRank) { |
| 756 | return UnknownShape(); |
| 757 | } |
| 758 | CHECK_GE(rank, 0) << "rank must not be negative"; |
| 759 | std::vector<DimensionHandle> dims(rank); |
| 760 | for (int32 i = 0; i < rank; ++i) { |
| 761 | dims[i] = UnknownDim(); |
| 762 | } |
| 763 | return MakeShape(dims); |
| 764 | } |
| 765 | |
| 766 | ShapeHandle InferenceContext::Scalar() { return MakeShape({}); } |
| 767 |