experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Replace operands now injected in OpBuilder

Browse Source 
These are now injected automatically.

PiperOrigin-RevId: 336357395
Change-Id: I41265f6ac6b049bc56b4fd48e28446f12e10c4f2
This commit is contained in:
Jacques Pienaar 2020-10-09 13:44:25 -07:00 committed by TensorFlower Gardener
parent fa513b979d
commit 776ec418d8
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/compiler/mlir/tfr/ir/tfr_ops.td
View File

@ -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 = [{