Searched refs:use_dataset (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | collective_all_reduce_strategy_test.py | 367 mode=['graph'], required_gpus=[0, 1, 2], use_dataset=[True, False]))
368 def testMakeInputFnIterator(self, required_gpus, use_dataset): argument
370 if use_dataset:
393 test_reinitialize=use_dataset,
394 ignore_order=not use_dataset)
511 use_dataset=[True, False],
513 def testMakeInputFnIterator(self, required_gpus, use_dataset, argument
515 if use_dataset:
538 test_reinitialize=use_dataset,
539 ignore_order=not use_dataset)
|
| D | parameter_server_strategy_test.py | 636 mode=['graph'], required_gpus=[1, 2], use_dataset=[True, False]))
637 def testMakeInputFnIteratorDistributed(self, required_gpus, use_dataset): argument
638 if use_dataset:
661 test_reinitialize=use_dataset,
662 ignore_order=not use_dataset)
666 mode=['graph'], required_gpus=[1, 2], use_dataset=[True, False]))
667 def testMakeInputFnIteratorLocal(self, required_gpus, use_dataset): argument
668 if use_dataset:
692 test_reinitialize=use_dataset,
693 ignore_order=not use_dataset)
|