Home
last modified time | relevance | path

Searched refs:ENABLE_CONTROL_FLOW_V2 (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_ops_benchmark.py94 old_val = control_flow_util.ENABLE_CONTROL_FLOW_V2
95 control_flow_util.ENABLE_CONTROL_FLOW_V2 = False
97 control_flow_util.ENABLE_CONTROL_FLOW_V2 = old_val
100 old_val = control_flow_util.ENABLE_CONTROL_FLOW_V2
101 control_flow_util.ENABLE_CONTROL_FLOW_V2 = True
103 control_flow_util.ENABLE_CONTROL_FLOW_V2 = old_val
106 old_val = control_flow_util.ENABLE_CONTROL_FLOW_V2
107 control_flow_util.ENABLE_CONTROL_FLOW_V2 = False
109 control_flow_util.ENABLE_CONTROL_FLOW_V2 = old_val
112 old_val = control_flow_util.ENABLE_CONTROL_FLOW_V2
[all …]
Dcontrol_flow_util.py28 ENABLE_CONTROL_FLOW_V2 = ((tf2.enabled() and variable
42 global ENABLE_CONTROL_FLOW_V2
43 ENABLE_CONTROL_FLOW_V2 = True
50 return ENABLE_CONTROL_FLOW_V2 or (
Dcontrol_flow_v2_toggles.py45 control_flow_util.ENABLE_CONTROL_FLOW_V2 = True
61 control_flow_util.ENABLE_CONTROL_FLOW_V2 = False
Dcontrol_flow_v2_enable_test.py30 self.assertTrue(control_flow_util.ENABLE_CONTROL_FLOW_V2)
Dcontrol_flow_v2_disable_test.py31 self.assertFalse(control_flow_util.ENABLE_CONTROL_FLOW_V2)
/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
Dcontrol_flow_util_v2_test.py30 self._enable_control_flow_v2_old = control_flow_util.ENABLE_CONTROL_FLOW_V2
31 control_flow_util.ENABLE_CONTROL_FLOW_V2 = True
34 control_flow_util.ENABLE_CONTROL_FLOW_V2 = self._enable_control_flow_v2_old
Dcontrol_flow_ops_py_test.py992 TypeError if control_flow_util.ENABLE_CONTROL_FLOW_V2 else ValueError,
993 v2_msg if control_flow_util.ENABLE_CONTROL_FLOW_V2 else v1_msg):
1250 if not control_flow_util.ENABLE_CONTROL_FLOW_V2:
1288 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
1439 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
1482 if not control_flow_util.ENABLE_CONTROL_FLOW_V2: return
1697 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
1760 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
1842 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
2659 if not control_flow_util.ENABLE_CONTROL_FLOW_V2:
[all …]
Dwhile_v2_test.py683 enter_count = 2 if control_flow_util.ENABLE_CONTROL_FLOW_V2 else 1
697 enter_count = 3 if control_flow_util.ENABLE_CONTROL_FLOW_V2 else 2
/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
Dtensor_array_ops_test.py398 if not control_flow_util.ENABLE_CONTROL_FLOW_V2:
498 if (control_flow_util.ENABLE_CONTROL_FLOW_V2 and
506 if (control_flow_util.ENABLE_CONTROL_FLOW_V2 and
524 not control_flow_util.ENABLE_CONTROL_FLOW_V2):
531 if (control_flow_util.ENABLE_CONTROL_FLOW_V2 and
540 if (control_flow_util.ENABLE_CONTROL_FLOW_V2 and
590 if control_flow_util.ENABLE_CONTROL_FLOW_V2 and
608 if control_flow_util.ENABLE_CONTROL_FLOW_V2 and
616 if control_flow_util.ENABLE_CONTROL_FLOW_V2 and not in_eager_mode:
626 if not control_flow_util.ENABLE_CONTROL_FLOW_V2 or in_eager_mode:
[all …]
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dmap_and_batch_test.py381 previous_control_flow_v2_value = control_flow_util.ENABLE_CONTROL_FLOW_V2
382 control_flow_util.ENABLE_CONTROL_FLOW_V2 = True
384 control_flow_util.ENABLE_CONTROL_FLOW_V2 = previous_control_flow_v2_value
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py559 enable_control_flow_v2_old = control_flow_util.ENABLE_CONTROL_FLOW_V2
560 control_flow_util.ENABLE_CONTROL_FLOW_V2 = True
564 control_flow_util.ENABLE_CONTROL_FLOW_V2 = enable_control_flow_v2_old
613 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
/external/tensorflow/tensorflow/compiler/tests/
Dtensor_array_ops_test.py1007 if not control_flow_util.ENABLE_CONTROL_FLOW_V2: