Searched defs:_in_multi_worker_mode (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | collective_all_reduce_strategy.py | 949 def _in_multi_worker_mode(self): member in CollectiveAllReduceExtended |
| H A D | one_device_strategy.py | 428 def _in_multi_worker_mode(self): member in OneDeviceExtended |
| H A D | parameter_server_strategy.py | 636 def _in_multi_worker_mode(self): member in ParameterServerStrategyExtended |
| H A D | mirrored_strategy.py | 899 def _in_multi_worker_mode(self): member in MirroredExtended |
| H A D | distribute_lib.py | 2723 def _in_multi_worker_mode(self): member in StrategyExtendedV2 |
| H A D | tpu_strategy.py | 1652 def _in_multi_worker_mode(self): member in TPUExtended |
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ | ||
| H A D | training.py | 2804 def _in_multi_worker_mode(self): member in Model |
| H A D | training_v1.py | 2800 def _in_multi_worker_mode(self): member in Model |