Searched refs:predict_batch_size (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_context.py | 200 predict_batch_size, argument
207 self._predict_batch_size = predict_batch_size
706 predict_batch_size, use_tpu): argument
710 predict_batch_size, use_tpu)
732 predict_batch_size, use_tpu, eval_on_tpu, argument
745 predict_batch_size, use_tpu)
748 predict_batch_size, use_tpu, eval_on_tpu,
|
| D | tpu_estimator.py | 2227 predict_batch_size=None, argument
2330 if predict_batch_size is not None:
2331 util_lib.check_positive_integer(predict_batch_size,
2361 self._config, train_batch_size, eval_batch_size, predict_batch_size,
|
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| D | tpu_gan_estimator_impl.py | 103 predict_batch_size=None, argument
257 predict_batch_size=predict_batch_size,
|
| D | tpu_gan_estimator_test.py | 175 predict_batch_size=8,
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | keras_correctness_test_base.py | 220 predict_batch_size = get_batch_size(get_data_size(x_predict),
223 predict_dataset = batch_wrapper(predict_dataset, predict_batch_size,
|
| D | keras_backward_compat_test.py | 278 predict_batch_size = len(x_predict)
280 predict_batch_size //= with_distribution.num_replicas_in_sync
283 predict_batch_size, with_distribution)
|