Searched defs:paddings_type (Results 1 – 6 of 6) sorted by relevance
129 mlir::RankedTensorType paddings_type = mlir::RankedTensorType::get( in PadInputOnUnshardedDim() local
864 auto paddings_type = op.paddings().getType().cast<ShapedType>(); in matchAndRewrite() local
1366 const TensorType paddings_type) { in SpaceToBatchNDBlockRank()1380 const auto paddings_type = op.paddings().getType().cast<TensorType>(); in verify() local
733 RankedTensorType paddings_type = in convertSpaceToBatchNDOp() local
1427 auto paddings_type = paddings.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
424 DataType paddings_type; member