Searched refs:eval_on_tpu (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| D | tpu_gan_estimator_impl.py | 105 eval_on_tpu=True, argument
235 is_on_tpu = _is_on_tpu(mode, use_tpu, eval_on_tpu)
259 eval_on_tpu=eval_on_tpu,
264 def _is_on_tpu(mode, use_tpu, eval_on_tpu): argument
268 return eval_on_tpu
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_context.py | 202 eval_on_tpu=True, argument
209 logging.info('_TPUContext: eval_on_tpu %s', eval_on_tpu)
210 if not use_tpu and eval_on_tpu:
213 self._eval_on_tpu = eval_on_tpu
732 predict_batch_size, use_tpu, eval_on_tpu, argument
748 predict_batch_size, use_tpu, eval_on_tpu,
|
| D | tpu_estimator.py | 2229 eval_on_tpu=True, argument
2362 use_tpu, eval_on_tpu, experimental_embedding_config_spec)
|