From 776ec418d83d8e746db315fb90dcfedf526a746f Mon Sep 17 00:00:00 2001
From: Jacques Pienaar <jpienaar@google.com>
Date: Fri, 9 Oct 2020 13:44:25 -0700
Subject: [PATCH] Replace operands now injected in OpBuilder

These are now injected automatically.

PiperOrigin-RevId: 336357395
Change-Id: I41265f6ac6b049bc56b4fd48e28446f12e10c4f2
---
 tensorflow/compiler/mlir/tfr/ir/tfr_ops.td | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/tensorflow/compiler/mlir/tfr/ir/tfr_ops.td b/tensorflow/compiler/mlir/tfr/ir/tfr_ops.td
index 2918336603b..562b3f79955 100644
--- a/tensorflow/compiler/mlir/tfr/ir/tfr_ops.td
+++ b/tensorflow/compiler/mlir/tfr/ir/tfr_ops.td
@@ -265,12 +265,11 @@ def TFR_ConstOp : TFR_Op<"constant", [ConstantLike, NoSideEffect]> {
 
   let hasFolder = 1;
 
-  let builders = [OpBuilder<
-    "OpBuilder &, OperationState &state, Attribute value",
+  let builders = [OpBuilder<"Attribute value",
     [{
       auto* ctx = value.getContext();
-      state.addAttribute("value", value);
-      state.addTypes(TFRAttrType::get(ctx));
+      $_state.addAttribute("value", value);
+      $_state.addTypes(TFRAttrType::get(ctx));
     }]>
   ];
 
@@ -400,8 +399,8 @@ def TFR_TFRFuncOp : TFR_Op<"func", [HasParent<"ModuleOp">,
   let skipDefaultBuilders = 1;
 
   let builders = [
-    OpBuilder<"OpBuilder &builder, OperationState &result, StringRef name, "
-              "FunctionType type, ArrayRef<NamedAttribute> attrs = {}">
+    OpBuilder<"StringRef name, FunctionType type, "
+              "ArrayRef<NamedAttribute> attrs = {}">
   ];
 
   let extraClassDeclaration = [{