Searched refs:paddings_type (Results 1 – 3 of 3) sorted by relevance
1186 const TensorType paddings_type) { in SpaceToBatchNDBlockRank() argument1189 } else if (paddings_type.hasStaticShape()) { in SpaceToBatchNDBlockRank()1190 return paddings_type.getShape()[0]; in SpaceToBatchNDBlockRank()1199 const auto paddings_type = op.paddings().getType().cast<TensorType>(); in Verify() local1210 << "requires rank of paddings = 2; got " << paddings_type.getRank(); in Verify()1214 if (paddings_type.hasStaticShape() && paddings_type.getShape()[1] != 2) { in Verify()1216 << paddings_type.getShape()[1]; in Verify()1220 if (block_shape_type.hasStaticShape() && paddings_type.hasStaticShape() && in Verify()1221 block_shape_type.getShape()[0] != paddings_type.getShape()[0]) { in Verify()1227 SpaceToBatchNDBlockRank(block_shape_type, paddings_type); in Verify()[all …]
833 auto paddings_type = op.paddings().getType().cast<ShapedType>(); in matchAndRewrite() local834 if (!paddings_type.hasRank()) { in matchAndRewrite()851 auto paddings_i64_type = RankedTensorType::get(paddings_type.getShape(), in matchAndRewrite()
795 RankedTensorType paddings_type = in convertSpaceToBatchNDOp() local811 if (!paddings_type) { in convertSpaceToBatchNDOp()