Searched refs:_get_tpu_context (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ | ||
| D | tpu_context.py | 731 def _get_tpu_context(config, train_batch_size, eval_batch_size, function |
| D | tpu_estimator.py | 2360 self._ctx = tpu_context._get_tpu_context( |