Searched refs:_replica_ctx_all_reduce (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_common_test.py | 135 reduced = strategy.extended._replica_ctx_all_reduce('SUM', value)
156 reduced = strategy.extended._replica_ctx_all_reduce('SUM', value)
|
| D | collective_all_reduce_strategy.py | 775 def _replica_ctx_all_reduce(self, reduce_op, value, options=None): member in CollectiveAllReduceExtended
787 return super()._replica_ctx_all_reduce(reduce_op, value, options)
|
| D | distribute_lib.py | 2390 def _replica_ctx_all_reduce(self, reduce_op, value, options=None): member in StrategyExtendedV2
|