Searched defs:_replica_ctx_all_reduce (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | collective_all_reduce_strategy.py | 833 def _replica_ctx_all_reduce(self, reduce_op, value, options=None): member in CollectiveAllReduceExtended |
| H A D | mirrored_strategy.py | 785 def _replica_ctx_all_reduce(self, reduce_op, value, options=None): member in MirroredExtended |
| H A D | distribute_lib.py | 2467 def _replica_ctx_all_reduce(self, reduce_op, value, options=None): member in StrategyExtendedV2 |