Searched refs:get_unscaled_gradients (Results 1 – 8 of 8) sorted by relevance
117 grads = opt.get_unscaled_gradients(scaled_grads)
650 def get_unscaled_gradients(self, grads): member in LossScaleOptimizer688 unscaled_grads = self.get_unscaled_gradients(grads)694 return self.get_unscaled_gradients(grads)
156 grads = opt.get_unscaled_gradients(scaled_grads)169 sparse_grad = opt.get_unscaled_gradients([sparse_scaled_grad])[0]
93 name: "get_unscaled_gradients"
94 name: "get_unscaled_gradients"
278 grads = model.optimizer.get_unscaled_gradients(grads)