Function _is_on_tpu

tensorflow/contrib/layers/python/layers/rev_block_lib.py:516–518 · view source on GitHub ↗

()

Source from the content-addressed store, hash-verified

514
515
516	def _is_on_tpu():
517	ctxt = framework_ops.get_default_graph()._get_control_flow_context() # pylint: disable=protected-access
518	return control_flow_util.GetContainingXLAContext(ctxt) is not None
519
520
521	def _recomputing_grad_fn(compute_fn,

_recompute_gradFunction · 0.85

_get_control_flow_contextMethod · 0.45

no test coverage detected