Searched refs:is_running_on_cpu (Results 1 – 2 of 2) sorted by relevance
415 def is_running_on_cpu(self, is_export_mode=False): member in _InternalTPUContext435 is_running_on_cpu = self._is_running_on_cpu(is_export_mode)436 if not is_running_on_cpu:438 return is_running_on_cpu472 if (self.is_running_on_cpu() or self.is_input_broadcast_with_iterators()):487 if (self.is_running_on_cpu() or self.is_input_broadcast_with_iterators()):
1643 if is_export_mode or self._ctx.is_running_on_cpu(is_export_mode):1696 running_on_cpu = self._ctx.is_running_on_cpu(is_export_mode)2589 if ctx.is_running_on_cpu():2611 if ctx.is_running_on_cpu():2668 if ctx.is_running_on_cpu(is_export_mode=False):2804 if ctx.is_running_on_cpu(is_export_mode=is_export_mode):