Home
last modified time | relevance | path

Searched refs:get_dataset (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/data/kernel_tests/
Dmemory_cleanup_test.py107 def _testIteratorMemoryLeak(self, get_dataset): argument
110 get_next = self.getNext(get_dataset())
126 def get_dataset(): function
133 self._testIteratorMemoryLeak(get_dataset)
138 def get_dataset(): function
145 self._testIteratorMemoryLeak(get_dataset)
150 def get_dataset(): function
157 self._testIteratorMemoryLeak(get_dataset)
162 def get_dataset(): function
169 self._testIteratorMemoryLeak(get_dataset)
[all …]
/external/tensorflow/tensorflow/python/keras/distribute/
Dkeras_premade_models_test.py60 def get_dataset(): function
79 hist = model.fit(get_dataset(), epochs=5)
97 hist = wide_deep_model.fit(get_dataset(), epochs=5)
Dkeras_utils_test.py90 dataset = keras_test_lib.get_dataset(distribution)
144 dataset = keras_test_lib.get_dataset(distribution)
168 dataset = keras_test_lib.get_dataset(distribution)
253 dataset = keras_test_lib.get_dataset(distribution)
449 dataset = keras_test_lib.get_dataset(distribution)
481 dataset = keras_test_lib.get_dataset(distribution)
Ddistribute_strategy_test.py194 def get_dataset(distribution): function
899 dataset = get_dataset(distribution)
937 dataset = get_dataset(distribution)
1238 dataset = get_dataset(distribution)
1256 dataset = get_dataset(distribution)
1387 dataset = get_dataset(distribution)
/external/tensorflow/tensorflow/python/keras/engine/
Ddata_adapter.py143 def get_dataset(self): member in DataAdapter
402 def get_dataset(self): member in TensorLikeDataAdapter
539 def get_dataset(self): member in DatasetCreatorAdapter
623 def get_dataset(self): member in CompositeTensorDataAdapter
687 def get_dataset(self): member in ListsOfScalarsDataAdapter
688 return self._internal_adapter.get_dataset()
730 def get_dataset(self): member in DatasetAdapter
887 def get_dataset(self): member in GeneratorDataAdapter
1191 dataset = self._adapter.get_dataset()
Ddata_adapter_test.py151 ds_iter = iter(adapter.get_dataset())
288 ds_iter = iter(adapter.get_dataset())
321 ds_iter = iter(adapter.get_dataset())
444 ds_iter = iter(adapter.get_dataset())
519 ds_iter = iter(adapter.get_dataset())
552 ds_iter = iter(adapter.get_dataset())
637 dataset = self.adapter_cls(self.dataset_input).get_dataset()
730 for i, data in enumerate(adapter.get_dataset()):
/external/tensorflow/tensorflow/python/keras/tests/
Dautomatic_outside_compilation_test.py222 dataset = distribute_strategy_test.get_dataset(strategy)
274 distribute_strategy_test.get_dataset(strategy))