Searched defs:_make_input_fn_iterator (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | collective_all_reduce_strategy.py | 704 def _make_input_fn_iterator( member in CollectiveAllReduceExtended |
| H A D | one_device_strategy.py | 294 def _make_input_fn_iterator( member in OneDeviceExtended |
| H A D | parameter_server_strategy.py | 354 def _make_input_fn_iterator( member in ParameterServerStrategyExtended |
| H A D | distribute_lib_test.py | 80 def _make_input_fn_iterator( member in _TestExtended |
| H A D | mirrored_strategy.py | 549 def _make_input_fn_iterator( member in MirroredExtended |
| H A D | distribute_lib.py | 2284 def _make_input_fn_iterator(self, input_fn, replication_mode): member in StrategyExtendedV2 |
| H A D | tpu_strategy.py | 944 def _make_input_fn_iterator( member in TPUExtended |