Home
last modified time | relevance | path

Searched refs:is_mlir_bridge_enabled (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dis_mlir_bridge_test_true.py28 def is_mlir_bridge_enabled(): function
Dis_mlir_bridge_test_false.py28 def is_mlir_bridge_enabled(): function
Dtest_util.py110 def is_mlir_bridge_enabled(): function
115 …from tensorflow.python.framework.is_mlir_bridge_test_false import is_mlir_bridge_enabled # pylint…
118 …from tensorflow.python.framework.is_mlir_bridge_test_true import is_mlir_bridge_enabled # pylint:…
1833 execute_func = not is_mlir_bridge_enabled()
2038 if is_mlir_bridge_enabled():
2040 elif is_mlir_bridge_enabled() is not None:
/external/tensorflow/tensorflow/compiler/tests/
Dxla_test.py92 if test_util.is_mlir_bridge_enabled():
94 elif test_util.is_mlir_bridge_enabled() is not None:
Dreduce_ops_test.py69 if not test_util.is_mlir_bridge_enabled():
/external/tensorflow/tensorflow/python/eager/
Ddef_function_xla_jit_test.py149 if test_util.is_mlir_bridge_enabled() else 'not compilable'):
168 if test_util.is_mlir_bridge_enabled() else 'not compilable'):
384 if test_util.is_mlir_bridge_enabled() else 'not compilable'):
893 if test_util.is_mlir_bridge_enabled():
Dcontext.py963 is_mlir_bridge_enabled = pywrap_tfe.TF_IsMlirBridgeEnabled()
964 config.experimental.mlir_bridge_rollout = is_mlir_bridge_enabled
965 if (is_mlir_bridge_enabled ==