HomeSort by relevance Sort by last modified time
    Searched refs:is_running_on_cpu (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/python/tpu/
tpu_context.py 191 if ctx.is_running_on_cpu():
415 def is_running_on_cpu(self, is_export_mode=False): member in class:_InternalTPUContext
435 is_running_on_cpu = self._is_running_on_cpu(is_export_mode)
436 if not is_running_on_cpu:
438 return is_running_on_cpu
472 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()):
tpu_estimator.py     [all...]

Completed in 155 milliseconds