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

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

View File

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