Searched defs:_use_merge_call (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | collective_all_reduce_strategy.py | 345 def _use_merge_call(self): member in CollectiveAllReduceExtended |
| H A D | mirrored_strategy.py | 351 def _use_merge_call(self): member in MirroredExtended |
| H A D | distribute_lib.py | 2962 def _use_merge_call(self): member in StrategyExtendedV1 |