Searched refs:crops_type (Results 1 – 2 of 2) sorted by relevance
299 auto crops_type = op.crops().getType().cast<TensorType>(); in Verify() local300 if (crops_type.hasRank()) { in Verify()301 if (crops_type.getRank() != 2) in Verify()303 << "requires crops to be a 2D tensor, but got " << crops_type; in Verify()306 if (crops_type.isDynamicDim(dim)) return true; in Verify()307 return crops_type.getDimSize(dim) == size; in Verify()311 << "requires crops to be a tensor<2x2>, but got " << crops_type; in Verify()
1056 RankedTensorType crops_type = in convertBatchToSpaceNDOp() local1071 if (!crops_type) { in convertBatchToSpaceNDOp()1079 int crops_dims = crops_type.getShape()[0]; in convertBatchToSpaceNDOp()