Searched refs:gradient_multipliers (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | training.py | 326 def multiply_gradients(grads_and_vars, gradient_multipliers): argument
343 if not gradient_multipliers:
345 if not isinstance(gradient_multipliers, dict):
350 if var in gradient_multipliers or var.op.name in gradient_multipliers:
351 key = var if var in gradient_multipliers else var.op.name
357 gradient_multipliers[key], dtype=grad.dtype)
361 gradient_multipliers[key], dtype=grad.dtype)
|
| D | training_test.py | 364 gradient_multipliers = {var: gradient_multiplier for var in variables}
367 return training.multiply_gradients(grads, gradient_multipliers)
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| D | learning.py | 302 def multiply_gradients(grads_and_vars, gradient_multipliers): argument
319 if not gradient_multipliers:
321 if not isinstance(gradient_multipliers, dict):
326 if var in gradient_multipliers or var.op.name in gradient_multipliers:
327 key = var if var in gradient_multipliers else var.op.name
331 multiplier = gradient_multipliers[key]
384 gradient_multipliers=None, argument
418 if gradient_multipliers:
420 grads = multiply_gradients(grads, gradient_multipliers)
|
| D | learning_test.py | 129 gradient_multipliers = {variable: self._multiplier}
131 learning.multiply_gradients(grad_to_var, gradient_multipliers)
151 gradient_multipliers = {variable: self._multiplier}
153 learning.multiply_gradients(grad_to_var, gradient_multipliers)
159 gradient_multipliers = {variable: self._multiplier}
162 gradient_multipliers)
181 gradient_multipliers = {variable: self._multiplier}
184 gradient_multipliers)
202 gradient_multipliers = {variable: tensor_multiplier}
205 gradient_multipliers)
[all …]
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | optimizers.py | 62 gradient_multipliers=None, argument
249 if gradient_multipliers is not None:
250 gradients = _multiply_gradients(gradients, gradient_multipliers)
429 def _multiply_gradients(grads_and_vars, gradient_multipliers): argument
434 (var in gradient_multipliers or var.name in gradient_multipliers)):
435 key = var if var in gradient_multipliers else var.name
436 multiplier = gradient_multipliers[key]
|
| D | optimizers_test.py | 244 gradient_multipliers={var: 7.})
262 gradient_multipliers={var: v})
280 gradient_multipliers={var: v})
300 gradient_multipliers={unused_variable: 1.},
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | dnn_linear_combined.py | 128 gradient_multipliers = {}
139 gradient_multipliers[v] = lr_mult
140 return gradient_multipliers
339 … gradient_multipliers=_extract_embedding_lr_multipliers( # pylint: disable=protected-access
|
| D | dnn.py | 200 gradient_multipliers=(
|