Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistribute_lib.py970 return self._extended._make_dataset_iterator(dataset) # pylint: disable=protected-access
1901 return self._extended._make_dataset_iterator(dataset) # pylint: disable=protected-access
2281 def _make_dataset_iterator(self, dataset): member in StrategyExtendedV2
3663 def _make_dataset_iterator(self, dataset): member in _DefaultDistributionExtended
H A Done_device_strategy.py287 def _make_dataset_iterator(self, dataset): member in OneDeviceExtended
H A Dparameter_server_strategy.py347 def _make_dataset_iterator(self, dataset): member in ParameterServerStrategyExtended
H A Dmirrored_strategy.py542 def _make_dataset_iterator(self, dataset): member in MirroredExtended
H A Ddistribute_lib_test.py524 iterator = default_strategy.extended._make_dataset_iterator(dataset)
H A Dcollective_all_reduce_strategy.py694 def _make_dataset_iterator(self, dataset): member in CollectiveAllReduceExtended
H A Dtpu_strategy.py934 def _make_dataset_iterator(self, dataset): member in TPUExtended