Searched refs:is_export_mode (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_estimator.py | 1411 def call_without_tpu(self, features, labels, is_export_mode): argument
1412 return self._call_model_fn(features, labels, is_export_mode=is_export_mode)
1584 features, labels, is_export_mode=False)
1623 is_export_mode): argument
1643 if is_export_mode or self._ctx.is_running_on_cpu(is_export_mode):
1661 def _call_model_fn(self, features, labels, is_export_mode=False): argument
1663 self._validate_model_features_and_labels(features, labels, is_export_mode)
1688 if is_export_mode:
1696 running_on_cpu = self._ctx.is_running_on_cpu(is_export_mode)
1699 if not is_export_mode:
[all …]
|
| D | tpu_context.py | 415 def is_running_on_cpu(self, is_export_mode=False): argument
435 is_running_on_cpu = self._is_running_on_cpu(is_export_mode)
440 def _is_running_on_cpu(self, is_export_mode): argument
451 if is_export_mode:
|