Searched refs:_replica_ctx_update (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_strategy.py | 810 def _replica_ctx_update(self, var, fn, args, kwargs, group): member in MirroredExtended
812 return super()._replica_ctx_update(var, fn, args, kwargs, group)
|
| D | distribute_lib.py | 2502 def _replica_ctx_update(self, var, fn, args=(), kwargs=None, group=True): member in StrategyExtendedV2
2635 return self._replica_ctx_update(
3545 …return self._strategy.extended._replica_ctx_update(var, fn, args=args, kwargs=kwargs, group=group)…
|