Searched defs:_distributed_apply (Results 1 – 3 of 3) sorted by relevance
188 def _distributed_apply(self, member in MixedPrecisionLossScaleOptimizer
743 def _distributed_apply(self, member in Optimizer
682 def _distributed_apply(self, distribution, grads_and_vars, name, apply_state): member in OptimizerV2