Searched refs:_get_iterator (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | minimize_loss_test.py | 69 def _get_iterator(self, strategy, input_fn): member in MinimizeLossStepTest
104 iterator = self._get_iterator(distribution, dataset_fn)
141 iterator = self._get_iterator(distribution, dataset_fn)
199 iterator = self._get_iterator(distribution, dataset_fn)
267 iterator = self._get_iterator(distribution, dataset_fn)
374 iterator = self._get_iterator(distribution, dataset_fn)
468 iterator = self._get_iterator(distribution, dataset_fn)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 1180 return self._get_iterator()
1201 return self._get_iterator()
1203 def _get_iterator(self): member in DistributedDatasetV1
1223 return self._get_iterator()
1333 return self._get_iterator()
1343 return self._get_iterator()
1345 def _get_iterator(self): member in DistributedDatasetsFromFunctionV1
1364 return self._get_iterator()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_arrays_v1.py | 144 input_iterator = _get_iterator(inputs, model._distribution_strategy)
202 val_iterator = _get_iterator(val_inputs, model._distribution_strategy)
536 def _get_iterator(inputs, distribution_strategy=None): function
|