Home
last modified time | relevance | path

Searched refs:dataset_fn (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/distribute/python/
Dinput_lib_test.py38 def _create_iterator(self, input_type, dataset_fn, worker_device_pairs, argument
52 iterator = input_lib.InputFunctionIterator(dataset_fn, input_workers,
56 dataset_fn(distribute_lib.InputContext()), input_workers,
62 dataset_fn, argument
69 input_type, dataset_fn, worker_device_pairs, devices, split_batch_by)
107 dataset_fn = lambda _: dataset_ops.Dataset.range(10) function
111 self._test_iterator(input_type, dataset_fn, worker_device_pairs,
120 dataset_fn = lambda _: dataset_ops.Dataset.range(10) function
124 self._test_iterator(input_type, dataset_fn, worker_device_pairs,
134 def dataset_fn(ctx): function
[all …]
Dsingle_loss_example.py35 def dataset_fn(): function
47 dataset_fn, loss_fn, optimizer, distribution, iterations_per_step)
59 def dataset_fn(): function
86 return model_fn, dataset_fn, layer
96 def dataset_fn(): function
125 return model_fn, dataset_fn, batchnorm
Dstep_fn.py53 def __init__(self, dataset_fn, distribution): argument
55 self._iterator = distribution.make_input_fn_iterator(lambda _: dataset_fn())
88 def __init__(self, dataset_fn, loss_fn, optimizer, distribution, argument
90 super(StandardSingleLossStep, self).__init__(dataset_fn, distribution)
Dminimize_loss_test.py61 model_fn, dataset_fn, layer = minimize_loss_example(
70 iterator = self._get_iterator(distribution, dataset_fn)
99 model_fn, dataset_fn, layer = minimize_loss_example(
102 iterator = self._get_iterator(distribution, dataset_fn)
149 model_fn, dataset_fn, _ = minimize_loss_example(
161 iterator = self._get_iterator(distribution, dataset_fn)
214 model_fn, dataset_fn, batchnorm = batchnorm_example(
230 iterator = self._get_iterator(distribution, dataset_fn)
311 def dataset_fn(): function
322 iterator = self._get_iterator(distribution, dataset_fn)
[all …]
Doptimizer_v2_test.py42 model_fn, dataset_fn, layer = minimize_loss_example(
44 iterator = distribution.make_input_fn_iterator(lambda _: dataset_fn())
Done_device_strategy_test.py50 dataset_fn = lambda: dataset_ops.Dataset.range(10) function
53 dataset_fn,
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dparallel_interleave_benchmark.py57 def _benchmark(self, dataset_fn, iters, num_elements): argument
61 dataset = dataset_fn().with_options(options)
77 def dataset_fn(): function
81 self._benchmark(dataset_fn=dataset_fn, iters=10, num_elements=100)
86 def dataset_fn(): function
91 self._benchmark(dataset_fn=dataset_fn, iters=100, num_elements=1000)
96 def dataset_fn(): function
101 self._benchmark(dataset_fn=dataset_fn, iters=100, num_elements=1000)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Ddataset_test.py94 def testDatasetSimpleSourceInputs(self, dataset_fn, num_inputs=0): argument
95 self.assertLen(dataset_fn()._inputs(), num_inputs)
99 dataset_fn = dataset_ops.Dataset.from_sparse_tensor_slices(
104 self.assertEmpty(dataset_fn._inputs())
144 def testUnaryTransformationInputs(self, dataset_fn, input_dataset_fn): argument
146 self.assertEqual([input_dataset], dataset_fn(input_dataset)._inputs())
150 dataset_fn = self.make_apply_fn(dataset_ops.Dataset.range(0))
151 self.assertEqual([input_dataset], dataset_fn(input_dataset)._inputs())
164 dataset_fn = self.make_interleave_fn(*interleave_fn_args)
165 self.assertEqual([input_dataset], dataset_fn(input_dataset)._inputs())
[all …]
Dtf_record_dataset_test.py43 def dataset_fn(self, member in TFRecordDatasetTest
71 dataset = self.dataset_fn(self.test_filenames[0])
77 dataset = self.dataset_fn(self.test_filenames[1])
83 dataset = self.dataset_fn(self.test_filenames)
91 dataset = self.dataset_fn(self.test_filenames, num_epochs=10)
99 dataset = self.dataset_fn(
121 dataset = self.dataset_fn(zlib_files, compression_type="ZLIB")
136 dataset = self.dataset_fn(gzip_files, compression_type="GZIP")
Dtext_line_dataset_test.py82 def dataset_fn(filenames, num_epochs, batch_size=None): function
92 dataset_fn([test_filenames[0]], 1), expected_output=expected_output)
96 dataset_fn([test_filenames[1]], 1),
103 dataset_fn(test_filenames, 1), expected_output=expected_output)
109 dataset_fn(test_filenames, 10), expected_output=expected_output * 10)
113 dataset_fn(test_filenames, 10, 5),
Dreduce_test.py135 def dataset_fn(): function
143 _ = dataset_fn().reduce(np.int64(0), reduce_fn)
153 def dataset_fn(): function
162 _ = dataset_fn().reduce(np.int64(0), reduce_fn)
172 def dataset_fn(): function
185 _ = dataset_fn().reduce(np.int64(0), reduce1_fn)
186 _ = dataset_fn().reduce(np.int64(0), reduce2_fn)
Dfixed_length_record_dataset_test.py77 def dataset_fn(filenames, num_epochs, batch_size=None): function
90 dataset_fn([test_filenames[0]], 1),
97 dataset_fn([test_filenames[1]], 1),
108 dataset_fn(test_filenames, 1), expected_output=expected_output)
111 get_next = self.getNext(dataset_fn(test_filenames, 10))
120 get_next = self.getNext(dataset_fn(test_filenames, 10, self._num_records))
Dshuffle_test.py46 def dataset_fn(count=5, buffer_size=None, seed=0): function
60 get_next = self.getNext(dataset_fn())
69 get_next = self.getNext(dataset_fn(buffer_size=100, seed=37))
80 get_next = self.getNext(dataset_fn(buffer_size=100, seed=37))
90 get_next = self.getNext(dataset_fn(buffer_size=100, seed=137))
103 get_next = self.getNext(dataset_fn(buffer_size=2, seed=37))
113 get_next = self.getNext(dataset_fn(count=0, buffer_size=100, seed=37))
Dcache_test.py52 def dataset_fn(count=5, filename=None): function
62 dataset_ops.get_legacy_output_shapes(dataset_fn()))
64 get_next = self.getNext(dataset_fn())
75 get_next = self.getNext(dataset_fn(filename=self.cache_prefix))
85 get_next = self.getNext(dataset_fn(count=0, filename=self.cache_prefix))
96 dataset_fn(count=0, filename=self.cache_prefix + "nonsense"))
Dzip_test.py35 def dataset_fn(components): function
48 get_next = self.getNext(dataset_fn(equal_length_components))
59 get_next = self.getNext(dataset_fn(variable_length_components))
Dlist_files_test.py111 def dataset_fn(): function
115 dataset_fn(),
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dparallel_interleave_test.py59 def dataset_fn(self, input_values, cycle_length, block_length, sloppy, member in ParallelInterleaveTest
176 self.dataset_fn(
207 self.dataset_fn(
235 self.dataset_fn(
277 self.dataset_fn(
317 self.dataset_fn(
360 self.dataset_fn(
398 self.dataset_fn(
418 self.dataset_fn(
440 self.dataset_fn(
[all …]
Dstats_dataset_ops_test.py181 def dataset_fn(): function
187 dataset_fn, {self.regexForNodeName("ParallelMapDataset")},
194 def dataset_fn(): function
200 dataset_fn, {self.regexForNodeName("ParallelMapDataset")},
207 def dataset_fn(): function
219 dataset_fn, {self.regexForNodeName("ParallelInterleaveDatasetV2")}, 10,
224 def dataset_fn(): function
233 dataset_fn, {self.regexForNodeName("ExperimentalMapAndBatchDataset")},
403 def dataset_fn(): function
417 dataset_fn, {self.regexForNodeName("ExperimentalParseExampleDataset")},
[all …]
Dscan_test.py48 def dataset_fn(start, step, take): function
54 next_element = self.getNext(dataset_fn(start_val, step_val, take_val))
84 def dataset_fn(start, step, take): function
91 next_element = self.getNext(dataset_fn(start_val, step_val, take_val))
Ddense_to_sparse_batch_test.py91 def dataset_fn(input_tensor): function
96 get_next = self.getNext(dataset_fn([[1]]))
102 get_next = self.getNext(dataset_fn(np.int32(range(13))))
Dget_single_element_test.py83 def dataset_fn(): function
88 _ = get_single_element.get_single_element(dataset_fn())
Dmap_and_batch_test.py68 def dataset_fn(batch_size, count, numa_aware=numa_aware): function
84 dataset = dataset_fn(14, 28)
102 get_next = self.getNext(dataset_fn(8, 14))
122 self.assertDatasetProduces(dataset_fn(8, 0), expected_output=[])
126 self.assertDatasetProduces(dataset_fn(0, 14), expected_output=[])
Dstats_dataset_test_base.py88 dataset_fn, argument
95 dataset = dataset_fn()
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils_test.py142 def test_assert_not_batched(self, dataset_fn, expected_error=None): argument
144 training_utils.assert_not_batched(dataset_fn())
147 training_utils.assert_not_batched(dataset_fn())
189 def test_assert_not_shuffled(self, dataset_fn, expected_error=None): argument
191 training_utils.assert_not_shuffled(dataset_fn())
194 training_utils.assert_not_shuffled(dataset_fn())
/external/tensorflow/tensorflow/python/grappler/
Ddatasets_test.py315 def dataset_fn(n): function
318 return dataset_fn
349 def dataset_fn(n): function
352 return dataset_fn

12