MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / no_xla_auto_jit_impl

Function no_xla_auto_jit_impl

tensorflow/python/framework/test_util.py:1705–1722  ·  view source on GitHub ↗

This test is not intended to be run with XLA auto jit enabled.

(func)

Source from the content-addressed store, hash-verified

1703def no_xla_auto_jit(description): # pylint: disable=unused-argument
1704
1705 def no_xla_auto_jit_impl(func):
1706 """This test is not intended to be run with XLA auto jit enabled."""
1707
1708 def decorator(func):
1709
1710 def decorated(self, *args, **kwargs):
1711 if is_xla_enabled():
1712 # Skip test if using XLA is forced.
1713 return
1714 else:
1715 return func(self, *args, **kwargs)
1716
1717 return decorated
1718
1719 if func is not None:
1720 return decorator(func)
1721
1722 return decorator
1723
1724 return no_xla_auto_jit_impl
1725

Callers

nothing calls this directly

Calls 1

decoratorFunction · 0.70

Tested by

no test coverage detected