Searched refs:DecoupledWeightDecayExtension (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | weight_decay_optimizers.py | 33 class DecoupledWeightDecayExtension(object): class
94 super(DecoupledWeightDecayExtension, self).__init__(**kwargs)
131 return super(DecoupledWeightDecayExtension, self).minimize(
162 return super(DecoupledWeightDecayExtension, self).apply_gradients(
172 super(DecoupledWeightDecayExtension, self)._prepare()
189 return super(DecoupledWeightDecayExtension, self)._apply_dense(grad, var)
193 return super(DecoupledWeightDecayExtension, self)._resource_apply_dense(
200 return super(DecoupledWeightDecayExtension, self)._apply_sparse(
214 return super(DecoupledWeightDecayExtension, self)._resource_apply_sparse(
255 class OptimizerWithDecoupledWeightDecay(DecoupledWeightDecayExtension,
[all …]
|