Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_ops_benchmark.py98 old_val = control_flow_util.ENABLE_CONTROL_FLOW_V2
99 control_flow_util.ENABLE_CONTROL_FLOW_V2 = False
101 control_flow_util.ENABLE_CONTROL_FLOW_V2 = old_val
104 old_val = control_flow_util.ENABLE_CONTROL_FLOW_V2
105 control_flow_util.ENABLE_CONTROL_FLOW_V2 = True
107 control_flow_util.ENABLE_CONTROL_FLOW_V2 = old_val
110 old_val = control_flow_util.ENABLE_CONTROL_FLOW_V2
111 control_flow_util.ENABLE_CONTROL_FLOW_V2 = False
113 control_flow_util.ENABLE_CONTROL_FLOW_V2 = old_val
116 old_val = control_flow_util.ENABLE_CONTROL_FLOW_V2
[all …]
Dcontrol_flow_v2_toggles.py47 control_flow_util.ENABLE_CONTROL_FLOW_V2 = True
62 control_flow_util.ENABLE_CONTROL_FLOW_V2 = False
Dcontrol_flow_util.py32 ENABLE_CONTROL_FLOW_V2 = ((tf2.enabled() and variable
46 global ENABLE_CONTROL_FLOW_V2
47 ENABLE_CONTROL_FLOW_V2 = True
54 return ENABLE_CONTROL_FLOW_V2 or (
Dcontrol_flow_v2_enable_test.py34 self.assertTrue(control_flow_util.ENABLE_CONTROL_FLOW_V2)
Dcontrol_flow_v2_disable_test.py35 self.assertFalse(control_flow_util.ENABLE_CONTROL_FLOW_V2)
/external/tensorflow/tensorflow/python/kernel_tests/
Dcontrol_flow_util_v2_test.py34 self._enable_control_flow_v2_old = control_flow_util.ENABLE_CONTROL_FLOW_V2
35 control_flow_util.ENABLE_CONTROL_FLOW_V2 = True
38 control_flow_util.ENABLE_CONTROL_FLOW_V2 = self._enable_control_flow_v2_old
Dtensor_array_ops_test.py402 if not control_flow_util.ENABLE_CONTROL_FLOW_V2:
502 if (control_flow_util.ENABLE_CONTROL_FLOW_V2 and
510 if (control_flow_util.ENABLE_CONTROL_FLOW_V2 and
528 not control_flow_util.ENABLE_CONTROL_FLOW_V2):
535 if (control_flow_util.ENABLE_CONTROL_FLOW_V2 and
544 if (control_flow_util.ENABLE_CONTROL_FLOW_V2 and
594 if control_flow_util.ENABLE_CONTROL_FLOW_V2 and
612 if control_flow_util.ENABLE_CONTROL_FLOW_V2 and
620 if control_flow_util.ENABLE_CONTROL_FLOW_V2 and not in_eager_mode:
630 if not control_flow_util.ENABLE_CONTROL_FLOW_V2 or in_eager_mode:
[all …]
Dcontrol_flow_ops_py_test.py972 TypeError if control_flow_util.ENABLE_CONTROL_FLOW_V2 else ValueError,
973 v2_msg if control_flow_util.ENABLE_CONTROL_FLOW_V2 else v1_msg):
1245 if not control_flow_util.ENABLE_CONTROL_FLOW_V2:
1283 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
1424 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
1467 if not control_flow_util.ENABLE_CONTROL_FLOW_V2: return
1682 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
1745 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
1827 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
2644 if not control_flow_util.ENABLE_CONTROL_FLOW_V2:
[all …]
Dwhile_v2_test.py668 enter_count = 2 if control_flow_util.ENABLE_CONTROL_FLOW_V2 else 1
682 enter_count = 3 if control_flow_util.ENABLE_CONTROL_FLOW_V2 else 2
/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/lite/experimental/examples/lstm/
Drnn.py170 if not control_flow_util.ENABLE_CONTROL_FLOW_V2:
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py525 enable_control_flow_v2_old = control_flow_util.ENABLE_CONTROL_FLOW_V2
526 control_flow_util.ENABLE_CONTROL_FLOW_V2 = True
530 control_flow_util.ENABLE_CONTROL_FLOW_V2 = enable_control_flow_v2_old
579 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
/external/tensorflow/tensorflow/compiler/tests/
Dtensor_array_ops_test.py1011 if not control_flow_util.ENABLE_CONTROL_FLOW_V2: