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

Remove derived attributes from tf.IfRegion and tf.WhileRegion op.

Browse Source 
These derived attributes are only necessary for export, where these ops will be converted to the functional form prior.

PiperOrigin-RevId: 326478653
Change-Id: Id9b4188f77d0491a15cc671eb575aff8b7ab5ebe
This commit is contained in:
Andy Ly 2020-08-13 10:47:11 -07:00 committed by TensorFlower Gardener
parent f03dc8cff2
commit 3d87c9d297
1 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
View File

@ -338,10 +338,6 @@ else_branch: A region that computes the outputs of the op if cond = false.
Variadic<TF_Tensor>:$output
);
TF_DerivedOperandTypeAttr Tcond = TF_DerivedOperandTypeAttr<0>;
TF_DerivedOperandTypeListAttr Tin = TF_DerivedOperandTypeListAttr<1>;
TF_DerivedResultTypeListAttr Tout = TF_DerivedResultTypeListAttr<0>;
let regions = (region SizedRegion<1>:$then_branch, SizedRegion<1>:$else_branch);
let verifier = [{
@ -755,8 +751,6 @@ def TL_WhileRegionOp : TF_Op<"WhileRegion",
);
let results = (outs Variadic<AnyTensor>:$output);
TF_DerivedOperandTypeListAttr T = TF_DerivedOperandTypeListAttr<0>;
let regions = (region SizedRegion<1>:$cond, SizedRegion<1>:$body);
let verifier = [{ return Verify(*this); }];