Searched refs:_get_iterator (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | minimize_loss_test.py | 44 def _get_iterator(self, strategy, input_fn): member in MinimizeLossStepTest
70 iterator = self._get_iterator(distribution, dataset_fn)
102 iterator = self._get_iterator(distribution, dataset_fn)
161 iterator = self._get_iterator(distribution, dataset_fn)
230 iterator = self._get_iterator(distribution, dataset_fn)
322 iterator = self._get_iterator(distribution, dataset_fn)
413 iterator = self._get_iterator(distribution, dataset_fn)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_arrays.py | 138 input_iterator = _get_iterator(inputs, model._distribution_strategy)
189 val_iterator = _get_iterator(val_inputs, model._distribution_strategy)
502 def _get_iterator(inputs, distribution_strategy=None): function
|