Home
last modified time | relevance | path

Searched refs:batch_wrapper (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/distribute/python/
Dkeras_dnn_correctness_test.py114 batch_wrapper(train_dataset, batch_size, distribution))
154 batch_wrapper(dataset, 4, distribution))
162 batch_wrapper(dataset, 4, distribution))
Dkeras_backward_compat_test.py162 def batch_wrapper(dataset, batch_size, distribution, repeat=None): function
185 dataset = batch_wrapper(dataset, 10, distribution)
193 dataset = batch_wrapper(dataset, 10, distribution)
252 x = batch_wrapper(
267 x = batch_wrapper(eval_dataset, batch_size, with_distribution)
282 predict_dataset = batch_wrapper(predict_dataset,
899 dataset = batch_wrapper(dataset, 32, distribution)
903 predict_dataset = batch_wrapper(predict_dataset, 32, distribution)
944 train_dataset = batch_wrapper(train_dataset, batch_size, distribution)
969 dataset = batch_wrapper(dataset, 4, distribution)
[all …]
Dkeras_correctness_test_base.py122 def batch_wrapper(dataset, batch_size, distribution, repeat=None): function
195 x = batch_wrapper(train_dataset, batch_size, with_distribution,
209 x = batch_wrapper(eval_dataset, batch_size, with_distribution)
223 predict_dataset = batch_wrapper(predict_dataset, predict_batch_size,
Dkeras_utils_test.py475 dataset = keras_test_lib.batch_wrapper(dataset, 32, distribution)
479 predict_dataset = keras_test_lib.batch_wrapper(predict_dataset, 32,
Dkeras_test.py196 def batch_wrapper(dataset, batch_size, distribution, repeat=None): function
219 dataset = batch_wrapper(dataset, 10, distribution)
227 dataset = batch_wrapper(dataset, 10, distribution)