Home
last modified time | relevance | path

Searched refs:ENABLE_CONTROL_FLOW_V2 (Results 1 – 9 of 9) 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_util.py31 ENABLE_CONTROL_FLOW_V2 = (os.getenv("TF_ENABLE_CONTROL_FLOW_V2", "0") != "0" or variable
41 return ENABLE_CONTROL_FLOW_V2 or (
/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.py358 if not control_flow_util.ENABLE_CONTROL_FLOW_V2:
441 if (control_flow_util.ENABLE_CONTROL_FLOW_V2 and
452 if (control_flow_util.ENABLE_CONTROL_FLOW_V2 and
460 if (control_flow_util.ENABLE_CONTROL_FLOW_V2 and
478 not control_flow_util.ENABLE_CONTROL_FLOW_V2):
485 if (control_flow_util.ENABLE_CONTROL_FLOW_V2 and
494 if (control_flow_util.ENABLE_CONTROL_FLOW_V2 and
544 if control_flow_util.ENABLE_CONTROL_FLOW_V2 and
562 if control_flow_util.ENABLE_CONTROL_FLOW_V2 and
570 if control_flow_util.ENABLE_CONTROL_FLOW_V2 and not in_eager_mode:
[all …]
Dcontrol_flow_ops_py_test.py445 v2_msg if control_flow_util.ENABLE_CONTROL_FLOW_V2 else v1_msg):
833 TypeError if control_flow_util.ENABLE_CONTROL_FLOW_V2 else ValueError,
834 v2_msg if control_flow_util.ENABLE_CONTROL_FLOW_V2 else v1_msg):
992 if not control_flow_util.ENABLE_CONTROL_FLOW_V2:
1170 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
1212 if not control_flow_util.ENABLE_CONTROL_FLOW_V2: return
1403 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
1466 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
1549 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
2260 if not control_flow_util.ENABLE_CONTROL_FLOW_V2:
[all …]
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dmap_and_batch_test.py466 previous_control_flow_v2_value = control_flow_util.ENABLE_CONTROL_FLOW_V2
467 control_flow_util.ENABLE_CONTROL_FLOW_V2 = True
469 control_flow_util.ENABLE_CONTROL_FLOW_V2 = previous_control_flow_v2_value
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn.py167 if not control_flow_util.ENABLE_CONTROL_FLOW_V2:
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py425 enable_control_flow_v2_old = control_flow_util.ENABLE_CONTROL_FLOW_V2
426 control_flow_util.ENABLE_CONTROL_FLOW_V2 = True
430 control_flow_util.ENABLE_CONTROL_FLOW_V2 = enable_control_flow_v2_old
479 if control_flow_util.ENABLE_CONTROL_FLOW_V2:
/external/tensorflow/tensorflow/compiler/tests/
Dtensor_array_ops_test.py936 if not control_flow_util.ENABLE_CONTROL_FLOW_V2: