| /external/tensorflow/tensorflow/contrib/slim/python/slim/nets/ |
| D | vgg_test.py | 119 train_batch_size = 2
126 (train_batch_size, train_height, train_width, 3))
129 [train_batch_size, num_classes])
253 train_batch_size = 2
260 (train_batch_size, train_height, train_width, 3))
263 [train_batch_size, num_classes])
394 train_batch_size = 2
401 (train_batch_size, train_height, train_width, 3))
404 [train_batch_size, num_classes])
|
| D | overfeat_test.py | 110 train_batch_size = 2
117 (train_batch_size, train_height, train_width, 3))
120 [train_batch_size, num_classes])
|
| D | alexnet_test.py | 110 train_batch_size = 2
117 (train_batch_size, train_height, train_width, 3))
120 [train_batch_size, num_classes])
|
| D | inception_v1_test.py | 191 train_batch_size = 5
197 (train_batch_size, height, width, 3))
|
| D | inception_v2_test.py | 244 train_batch_size = 5
250 (train_batch_size, height, width, 3))
|
| D | inception_v3_test.py | 273 train_batch_size = 5
279 (train_batch_size, height, width, 3))
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_context.py | 198 train_batch_size, argument
205 self._train_batch_size = train_batch_size
705 def __init__(self, config, train_batch_size, eval_batch_size, argument
709 config, train_batch_size, eval_batch_size,
731 def _get_tpu_context(config, train_batch_size, eval_batch_size, argument
744 return _OneCoreTPUContext(config, train_batch_size, eval_batch_size,
747 return _InternalTPUContext(config, train_batch_size, eval_batch_size,
|
| D | _tpu_estimator_embedding.py | 213 def __init__(self, embedding_config_spec, train_batch_size, eval_batch_size, argument
216 self._train_batch_size = train_batch_size
|
| D | tpu_estimator.py | 2225 train_batch_size=None, argument
2316 if train_batch_size is None:
2318 util_lib.check_positive_integer(train_batch_size, 'train_batch_size')
2361 self._config, train_batch_size, eval_batch_size, predict_batch_size,
3796 train_batch_size=2048, # Does not matter.
|
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| D | tpu_gan_estimator_impl.py | 101 train_batch_size=None, argument
255 train_batch_size=train_batch_size,
|
| D | tpu_gan_estimator_test.py | 173 train_batch_size=4,
272 train_batch_size=4,
291 train_batch_size=4,
|
| /external/tensorflow/tensorflow/python/debug/examples/ |
| D | debug_mnist.py | 50 xs, ys = mnist.train.next_batch(FLAGS.train_batch_size,
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
| D | main_estimator_tpu.py | 222 train_batch_size=revnet_config.tpu_batch_size,
|
| /external/tensorflow/tensorflow/contrib/tpu/ |
| D | tpu_estimator.md | 112 train_batch_size=FLAGS.batch_size,
|