| 41 | } |
| 42 | |
| 43 | void TensorShape::CheckDimsEqual(int NDIMS) const { |
| 44 | CHECK_EQ(NDIMS, dims()) << "Asking for tensor of " << NDIMS << " dimensions" |
| 45 | << " from a tensor of " << dims() << " dimensions"; |
| 46 | } |
| 47 | |
| 48 | void TensorShape::CheckDimsAtLeast(int NDIMS) const { |
| 49 | CHECK_GE(NDIMS, dims()) << "Asking for tensor of at least " << NDIMS |