clean up expired compat check
PiperOrigin-RevId: 327888730 Change-Id: Ib16049bc731638dd2ce6ceff396fac2e2ca500a1
This commit is contained in:
parent
dcc2e62c8e
commit
90d58ce333
@ -20,7 +20,6 @@ from __future__ import division
|
|||||||
from __future__ import print_function
|
from __future__ import print_function
|
||||||
|
|
||||||
from tensorflow.core.protobuf import config_pb2
|
from tensorflow.core.protobuf import config_pb2
|
||||||
from tensorflow.python.compat.compat import forward_compatibility_horizon
|
|
||||||
from tensorflow.python.eager import backprop
|
from tensorflow.python.eager import backprop
|
||||||
from tensorflow.python.eager import context
|
from tensorflow.python.eager import context
|
||||||
from tensorflow.python.eager import def_function
|
from tensorflow.python.eager import def_function
|
||||||
@ -1607,5 +1606,4 @@ def _has_node_with_op(run_metadata, op_type):
|
|||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
with forward_compatibility_horizon(2020, 8, 21):
|
test.main()
|
||||||
test.main()
|
|
||||||
|
@ -25,7 +25,6 @@ from __future__ import print_function
|
|||||||
|
|
||||||
import collections
|
import collections
|
||||||
|
|
||||||
from tensorflow.python.compat import compat
|
|
||||||
from tensorflow.python.eager import backprop_util
|
from tensorflow.python.eager import backprop_util
|
||||||
from tensorflow.python.framework import auto_control_deps
|
from tensorflow.python.framework import auto_control_deps
|
||||||
from tensorflow.python.framework import auto_control_deps_utils as acd
|
from tensorflow.python.framework import auto_control_deps_utils as acd
|
||||||
@ -1120,10 +1119,7 @@ def _build_case(branch_index,
|
|||||||
op for op in bg.get_operations() if auto_control_deps.op_is_stateful(op)
|
op for op in bg.get_operations() if auto_control_deps.op_is_stateful(op)
|
||||||
])
|
])
|
||||||
|
|
||||||
# TODO(b/161915509): Remove this after 08/20/2020. This is required to abide
|
if stateful_ops:
|
||||||
# by 3-week forward compat window of new TF python op generating code with
|
|
||||||
# stale runtime binaries.
|
|
||||||
if (stateful_ops or not compat.forward_compatible(2020, 8, 20)):
|
|
||||||
op_fn = gen_functional_ops.case
|
op_fn = gen_functional_ops.case
|
||||||
else:
|
else:
|
||||||
op_fn = gen_functional_ops.stateless_case
|
op_fn = gen_functional_ops.stateless_case
|
||||||
|
Loading…
Reference in New Issue
Block a user