Searched refs:RewriteOp (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fold_broadcast.cc | 47 LogicalResult RewriteOp(
65 return RewriteOp(op, rewriter, OpTrait::util::getBroadcastedShape); in matchAndRewrite()
120 return RewriteOp(op, rewriter, get_broadcasted_shape); in RewriteBatchMatMulV2Op()
128 return RewriteOp(op, rewriter, OpTrait::util::getBroadcastedShape); in RewriteEqOp()
132 LogicalResult ConvertResultsBroadcastableShapeOp::RewriteOp( in RewriteOp() function in mlir::__anondc9c58b40111::ConvertResultsBroadcastableShapeOp
|
| /external/llvm-project/mlir/lib/Dialect/PDL/IR/ |
| D | PDL.cpp | 74 if (isa<AttributeOp, InputOp, OperationOp, RewriteOp>(user)) in verifyHasBindingUseInMatcher()
100 if (!attrValue && isa<RewriteOp>(op->getParentOp())) in verify()
276 bool isWithinRewrite = isa<RewriteOp>(op->getParentOp()); in verify()
327 if (!isa<RewriteOp>(term)) { in verify()
359 RewriteOp PatternOp::getRewriter() { in getRewriter()
360 return cast<RewriteOp>(body().front().getTerminator()); in getRewriter()
413 static LogicalResult verify(RewriteOp op) { in verify()
|
| /external/llvm-project/mlir/include/mlir/Dialect/PDL/IR/ |
| D | PDLOps.td | 121 : PDL_Op<"create_native", [HasParent<"pdl::RewriteOp">]> {
153 def PDL_EraseOp : PDL_Op<"erase", [HasParent<"pdl::RewriteOp">]> {
327 RewriteOp getRewriter();
340 AttrSizedOperandSegments, HasParent<"pdl::RewriteOp">
372 // pdl::RewriteOp
418 HasParent<"pdl::RewriteOp">]> {
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 302 static LogicalResult RewriteOp(Operation* op, PatternRewriter& rewriter, in RewriteOp() function in mlir::mhlo::__anon94f199800111::Tf2XlaRewriter
582 return Tf2XlaRewriter::RewriteOp(op, rewriter, device_type_); in matchAndRewrite()
|
| /external/llvm-project/mlir/test/Dialect/PDL/ |
| D | invalid.mlir | 196 // pdl::RewriteOp
|
| /external/llvm-project/mlir/lib/Conversion/PDLToPDLInterp/ |
| D | PDLToPDLInterp.cpp | 447 pdl::RewriteOp rewriter = pattern.getRewriter(); in generateRewriter()
|