| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | make_batched_features_dataset_test.py | 42 for num_epochs in [1, 10]:
48 num_epochs=num_epochs,
51 batch_size, 0, num_epochs=num_epochs, label_key_provided=True)
60 num_epochs=num_epochs,
63 batch_size, 1, num_epochs=num_epochs, label_key_provided=True)
72 num_epochs=num_epochs,
75 batch_size, num_epochs=num_epochs, label_key_provided=True)
82 num_epochs=num_epochs,
84 self.verify_records(batch_size, num_epochs=num_epochs)
107 num_epochs = 5
[all …]
|
| D | make_tf_record_dataset_test.py | 34 def _read_test(self, batch_size, num_epochs, file_index=None, argument
49 num_epochs=num_epochs,
59 num_epochs=num_epochs,
68 for num_epochs in [1, 3]:
70 self._read_test(batch_size, num_epochs, 0)
73 self._read_test(batch_size, num_epochs, 1)
76 self._read_test(batch_size, num_epochs)
79 self._read_test(batch_size, num_epochs, num_parallel_reads=8)
83 for num_epochs in [1, 3]:
85 self._read_test(batch_size, num_epochs, 0, drop_final_batch=True)
[all …]
|
| D | make_csv_dataset_test.py | 40 def _make_csv_dataset(self, filenames, batch_size, num_epochs=1, **kwargs): argument
42 filenames, batch_size=batch_size, num_epochs=num_epochs, **kwargs)
65 num_epochs): argument
67 for _ in range(num_epochs):
81 num_epochs, argument
92 num_epochs,
113 num_epochs=1, argument
123 num_epochs=num_epochs,
126 self._verify_output(dataset, batch_size, num_epochs, label_name,
154 num_epochs=1,
[all …]
|
| D | reader_dataset_ops_test_base.py | 74 num_epochs, argument
83 self.num_epochs = num_epochs
97 num_epochs=self.num_epochs,
195 num_epochs, argument
214 for _ in range(num_epochs):
252 num_epochs=1, argument
263 num_epochs,
336 def _next_expected_batch(self, file_indices, batch_size, num_epochs, argument
350 for _ in range(num_epochs):
368 def _verify_records(self, outputs, batch_size, file_index, num_epochs, argument
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | tf_record_dataset_serialization_test.py | 35 num_epochs, argument
63 buffer_size=buffer_size).repeat(num_epochs).batch(batch_size)
66 num_epochs = 5
67 batch_size = num_epochs
68 num_outputs = num_epochs * self._num_files * self._num_records // batch_size
71 lambda: self._build_iterator_graph(num_epochs, batch_size,
73 lambda: self._build_iterator_graph(num_epochs * 2, batch_size),
76 lambda: self._build_iterator_graph(num_epochs, buffer_size=0), None,
81 num_epochs = 5
82 num_outputs = num_epochs * self._num_files * self._num_records
[all …]
|
| D | fixed_length_record_dataset_serialization_test.py | 30 def _build_iterator_graph(self, num_epochs, compression_type=None): argument
34 self._footer_bytes).repeat(num_epochs)
37 num_epochs = 5
38 num_outputs = num_epochs * self._num_files * self._num_records
39 self.run_core_tests(lambda: self._build_iterator_graph(num_epochs),
40 lambda: self._build_iterator_graph(num_epochs * 2),
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/ |
| D | graph_io.py | 53 num_epochs=None, argument
106 num_epochs=num_epochs,
121 num_epochs=None, argument
176 num_epochs=num_epochs,
191 num_epochs=None, argument
249 num_epochs=num_epochs,
342 num_epochs=None, argument
401 if (num_epochs is not None) and (num_epochs <= 0):
402 raise ValueError('Invalid num_epochs %s.' % num_epochs)
413 num_epochs=num_epochs,
[all …]
|
| D | pandas_io_test.py | 63 x, y_noindex, batch_size=2, shuffle=False, num_epochs=1)
71 x, y, batch_size=2, shuffle=False, num_epochs=1)
89 x, y, batch_size=128, shuffle=False, num_epochs=2)
118 x, y, batch_size=2, shuffle=False, num_epochs=1)
152 x, y=None, batch_size=2, shuffle=False, num_epochs=1)
165 x, y, batch_size=2, shuffle=False, num_epochs=1)
188 x, y, batch_size=4, shuffle=False, num_epochs=1)
198 x, y, batch_size=4, shuffle=True, num_epochs=1)
208 x, y, batch_size=2, shuffle=True, queue_capacity=None, num_epochs=2)
218 x, y, batch_size=3, shuffle=False, num_epochs=1)
[all …]
|
| D | generator_io_test.py | 47 num_epochs=1)
73 generator, target_key=None, batch_size=2, shuffle=False, num_epochs=1)
106 num_epochs=1)
143 num_epochs=1)
168 x, batch_size=2, shuffle=False, num_epochs=1)
179 generator, batch_size=2, shuffle=False, num_epochs=1)
190 generator, batch_size=2, shuffle=False, num_epochs=1)
208 generator, target_key=y, batch_size=2, shuffle=False, num_epochs=1)
226 generator, target_key=y, batch_size=2, shuffle=False, num_epochs=1)
243 generator, target_key=y, batch_size=2, shuffle=False, num_epochs=1)
[all …]
|
| D | graph_io_test.py | 86 num_epochs=None,
98 num_epochs=None,
110 num_epochs=None,
122 num_epochs=None,
134 num_epochs=None,
146 num_epochs=None,
158 num_epochs=None,
170 num_epochs=None,
182 num_epochs=-1,
194 num_epochs=None,
[all …]
|
| D | generator_io.py | 38 num_epochs=1, argument
130 num_epochs=num_epochs,
134 if num_epochs is None else queue.dequeue_up_to(batch_size))
|
| D | numpy_io.py | 34 num_epochs=1, argument
43 num_epochs=num_epochs,
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | dnn_test.py | 253 def _input_fn_float_label(num_epochs=None): argument
258 ['en', 'fr', 'zh'], num_epochs=num_epochs),
367 def _input_fn(num_epochs=None): argument
372 num_epochs=num_epochs),
376 ['en', 'fr', 'zh'], num_epochs=num_epochs),
400 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
411 def _input_fn(num_epochs=None): argument
416 num_epochs=num_epochs),
420 ['en', 'fr', 'zh'], num_epochs=num_epochs),
445 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
[all …]
|
| D | debug_test.py | 273 def _input_fn(num_epochs=None): argument
278 num_epochs=num_epochs),
282 ['en', 'fr', 'zh'], num_epochs=num_epochs),
295 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
302 def _input_fn_float_label(num_epochs=None): argument
307 num_epochs=num_epochs),
311 ['en', 'fr', 'zh'], num_epochs=num_epochs),
322 predict_input_fn = functools.partial(_input_fn_float_label, num_epochs=1)
463 def _input_fn(num_epochs=None): argument
470 num_epochs=num_epochs),
[all …]
|
| D | dnn_linear_combined_test.py | 314 def _input_fn_float_label(num_epochs=None): argument
319 ['en', 'fr', 'zh'], num_epochs=num_epochs),
548 def _input_fn(num_epochs=None): argument
553 num_epochs=num_epochs),
557 ['en', 'fr', 'zh'], num_epochs=num_epochs),
585 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
776 shape=[4, 1], dtype=dtypes.float32), num_epochs=1)
795 def _input_fn(num_epochs=None): argument
803 num_epochs=num_epochs)
840 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
[all …]
|
| D | linear_test.py | 182 def _input_fn(num_epochs=None): argument
187 ['en', 'fr', 'zh'], num_epochs=num_epochs),
209 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
242 def _input_fn(num_epochs=None): argument
247 num_epochs=num_epochs),
251 ['en', 'fr', 'zh'], num_epochs=num_epochs),
363 def _input_fn(num_epochs=None): argument
371 num_epochs=num_epochs)
405 predict_input_fn = functools.partial(_input_fn, num_epochs=1)
446 def input_fn(num_epochs=None): argument
[all …]
|
| /external/tensorflow/tensorflow/python/data/benchmarks/ |
| D | from_tensor_slices_benchmark.py | 33 num_epochs = 100
34 num_elements = input_size * num_epochs // batch_size
40 num_epochs).batch(batch_size))
50 num_epochs = 100
52 num_elements = num_epochs * reshape_dim[0]
58 input_data.reshape(*reshape_dim)).repeat(num_epochs))
69 num_epochs = 100
70 num_elements = input_size * num_epochs // batch_size
76 batch_size).cache().repeat(num_epochs))
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/linear_regression/ |
| D | linear_regression_graph_test.py | 29 num_epochs = 10
59 def train(num_epochs): argument
60 for _ in range(num_epochs):
72 train(num_epochs)
75 examples_per_sec = num_epochs * num_batches * batch_size / wall_time
79 iters=num_epochs * num_batches,
|
| /external/tensorflow/tensorflow/contrib/kafka/python/kernel_tests/ |
| D | kafka_test.py | 46 num_epochs = array_ops.placeholder(dtypes.int64, shape=[])
50 topics, group="test", eof=True).repeat(num_epochs)
61 sess.run(init_op, feed_dict={topics: ["test:0:0:4"], num_epochs: 1})
68 sess.run(init_op, feed_dict={topics: ["test:0:5:-1"], num_epochs: 1})
79 num_epochs: 1
92 num_epochs: 10
106 num_epochs: 10,
|
| /external/tensorflow/tensorflow/contrib/input_pipeline/python/ops/ |
| D | input_pipeline_ops.py | 55 def _create_list(string_list, shuffle, seed, num_epochs): argument
59 if num_epochs:
60 for _ in range(num_epochs - 1):
65 def seek_next(string_list, shuffle=False, seed=None, num_epochs=None): argument
85 expanded_list = _create_list(string_list, shuffle, seed, num_epochs)
100 if num_epochs:
|
| /external/tensorflow/tensorflow/examples/how_tos/reading_data/ |
| D | fully_connected_reader.py | 84 def inputs(train, batch_size, num_epochs): argument
104 if not num_epochs:
105 num_epochs = None
126 dataset = dataset.repeat(num_epochs)
140 train=True, batch_size=FLAGS.batch_size, num_epochs=FLAGS.num_epochs)
181 print('Done training for %d epochs, %d steps.' % (FLAGS.num_epochs,
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | readers.py | 44 num_epochs=None, argument
140 num_epochs, shuffle, shuffle_buffer_size, shuffle_seed,
172 num_epochs=None, argument
272 num_epochs, shuffle, shuffle_buffer_size, shuffle_seed,
285 num_epochs=None, argument
355 num_epochs=num_epochs,
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | dataset_checkpoint_test.py | 115 def _build_graph(start, stop, num_epochs): argument
116 dataset = dataset_ops.Dataset.range(start, stop).repeat(num_epochs)
127 num_epochs = 5
131 init_op, get_next, save_op, _ = _build_graph(start, stop, num_epochs)
154 for _ in range(break_epoch + 1, num_epochs):
278 def _build_graph(start, stop, num_epochs): argument
280 dataset_ops.Dataset.range(start, stop).repeat(num_epochs))
289 num_epochs = 5
294 start, stop, num_epochs)
310 init_op, get_next, _, restore_op = _build_graph(start, stop, num_epochs)
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | readers.py | 206 dataset, num_epochs, shuffle, shuffle_buffer_size, shuffle_seed): argument
208 if num_epochs != 1 and shuffle:
211 shuffle_ops.shuffle_and_repeat(shuffle_buffer_size, num_epochs,
215 elif num_epochs != 1:
216 return dataset.repeat(num_epochs)
223 num_epochs=None, argument
288 dataset, num_epochs, shuffle, shuffle_buffer_size, shuffle_seed)
294 drop_final_batch = drop_final_batch or num_epochs is None
323 num_epochs=None, argument
502 dataset, num_epochs, shuffle, shuffle_buffer_size, shuffle_seed)
[all …]
|
| /external/tensorflow/tensorflow/contrib/kinesis/python/kernel_tests/ |
| D | kinesis_test.py | 56 num_epochs = array_ops.placeholder(dtypes.int64, shape=[])
60 stream, read_indefinitely=False).repeat(num_epochs)
71 sess.run(init_op, feed_dict={stream: stream_name, num_epochs: 1})
100 num_epochs = array_ops.placeholder(dtypes.int64, shape=[])
104 stream, shard, read_indefinitely=False).repeat(num_epochs)
118 stream: stream_name, shard: shard_id_0, num_epochs: 1})
127 stream: stream_name, shard: shard_id_1, num_epochs: 1})
|