Searched refs:KerasCrossShardOptimizer (Results 1 – 1 of 1) sorted by relevance
326 class KerasCrossShardOptimizer(keras_optimizers.Optimizer): class340 super(KerasCrossShardOptimizer, self).__init__()351 grads = super(KerasCrossShardOptimizer, self).get_gradients(loss, params)385 return KerasCrossShardOptimizer(opt)