Home
last modified time | relevance | path

Searched defs:num_epochs (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dmake_batched_features_dataset_test.py41 def testRead(self, batch_size, num_epochs): argument
111 num_epochs): argument
139 self, batch_size, num_epochs): argument
172 def testParallelReadersAndParsers(self, batch_size, num_epochs, argument
208 def testDropFinalBatch(self, batch_size, num_epochs): argument
Dmake_tf_record_dataset_test.py42 def testRead(self, batch_size, num_epochs, file_index, num_parallel_reads, argument
82 def testShuffle(self, batch_size, num_epochs, num_parallel_reads, seed): argument
Dmake_csv_dataset_test.py36 def _make_csv_dataset(self, filenames, batch_size, num_epochs=1, **kwargs): argument
65 num_epochs): argument
81 num_epochs, argument
113 num_epochs=1, argument
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dtf_record_test_base.py41 num_epochs, argument
162 num_epochs, argument
219 num_epochs=1, argument
269 def _next_expected_batch(self, file_indices, batch_size, num_epochs, argument
301 def _verify_records(self, outputs, batch_size, file_index, num_epochs, argument
Dfixed_length_record_dataset_test.py81 def dataset_fn(filenames, num_epochs, batch_size=None): argument
233 def _build_dataset(self, num_epochs, compression_type=None): argument
Dtf_record_dataset_test.py39 num_epochs=1, argument
196 num_epochs, argument
Dcheckpoint_test.py209 def _build_graph(start, stop, num_epochs): argument
257 def _build_graph(start, stop, num_epochs): argument
Dtext_line_dataset_test.py90 def dataset_fn(filenames, num_epochs, batch_size=None): argument
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dreaders.py218 dataset, num_epochs, shuffle, shuffle_buffer_size, shuffle_seed): argument
230 num_epochs=None, argument
337 num_epochs=None, # TODO(aaudibert): Change default to 1 when graduating. argument
641 num_epochs=None, argument
922 num_epochs=None, argument
1106 num_epochs=None, argument
/external/tensorflow/tensorflow/python/training/
Dinput.py80 def limit_epochs(tensor, num_epochs=None, name=None): argument
120 num_epochs=None, argument
207 num_epochs=None, argument
280 def range_input_producer(limit, num_epochs=None, shuffle=True, seed=None, argument
323 def slice_input_producer(tensor_list, num_epochs=None, shuffle=True, seed=None, argument
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py190 def _Run(self, is_training, use_trt, batch_size, num_epochs, model_dir): argument
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc890 const int num_epochs = 100; in AddControlEdges() local