Searched refs:is_tfrt_enabled (Results 1 – 8 of 8) sorted by relevance
108 name: "is_tfrt_enabled"
142 success = _shutdown_tpu_system() if context.is_tfrt_enabled() else True811 was_enabled = context.is_tfrt_enabled()
89 if context.is_tfrt_enabled():
76 is_tfrt_enabled = tfrt_utils.enabled variable115 tf_export("__internal__.is_tfrt_enabled", v1=[])(is_tfrt_enabled)464 self._use_tfrt = is_tfrt_enabled()
138 if context.is_tfrt_enabled():
414 if context.is_tfrt_enabled():
347 if context.is_tfrt_enabled():
1962 if context.is_tfrt_enabled():