Searched refs:tfrt (Results 1 – 6 of 6) sorted by relevance
1694 def use_tfrt(self, tfrt): argument1696 if not isinstance(tfrt, bool):1697 raise ValueError("Expecting a boolean but got %s" % type(tfrt))1699 if self._use_tfrt != tfrt:1702 self._use_tfrt = tfrt
118 tfrt::tf::ContextInterface* tfrt_context = new tfrt::tf::ContextInterface( in TFE_NewContext()125 tfrt::tf::CreateDistributedManagerContext( in TFE_NewContext()
606 void ExecuteAdd(bool async, bool forward_input, bool tfrt) { in ExecuteAdd() argument609 TFE_ContextOptionsSetTfrt(opts, tfrt); in ExecuteAdd()
677 "//tensorflow/core/tfrt/utils:__subpackages__",768 "//tensorflow/core/tfrt/utils:__subpackages__",
21 "//learning/brain/experimental/tfrt/...",
2288 tags = tags + ["tfrt"],