Searched refs:_is_running_on_cpu (Results 1 – 3 of 3) sorted by relevance
286 if _is_running_on_cpu():384 if _is_running_on_cpu():433 def _is_running_on_cpu(): function
435 is_running_on_cpu = self._is_running_on_cpu(is_export_mode)440 def _is_running_on_cpu(self, is_export_mode): member in _InternalTPUContext
24 from tensorflow.python.tpu.feature_column import _is_running_on_cpu