Searched refs:stop_gradients (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_impl.py | 51 stop_gradients=None, argument
157 gate_gradients, aggregation_method, stop_gradients,
169 stop_gradients=None, argument
273 aggregation_method, stop_gradients,
|
| D | gradients_util.py | 553 stop_gradients=None, argument
582 stop_gradients = [] if stop_gradients is None else _AsList(stop_gradients)
590 list(ys) + list(xs) + list(stop_gradients) + list(grad_ys)) as grad_scope:
615 stop_gradient_ops = [t.op for t in stop_gradients]
|
| D | gradients_test.py | 311 partialg = gradients.gradients(z, [x, y], stop_gradients=[x, y])
316 def _MakeGraph(rng, stop_gradients=()): argument
323 if "a" in stop_gradients: a = array_ops.stop_gradient(a)
325 if "b" in stop_gradients: b = array_ops.stop_gradient(b)
327 if "c" in stop_gradients: c = array_ops.stop_gradient(c)
329 if "d" in stop_gradients: d = array_ops.stop_gradient(d)
351 stop_gradients=[graph[v] for v in constants])
|
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | optimizer_v2.py | 662 stop_gradients=None, argument
713 stop_gradients=stop_gradients,
732 stop_gradients=None, argument
824 stop_gradients=stop_gradients)
|
| D | optimizer_v2_test.py | 257 stop_gradients=[var0_id])
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 645 …\'grad_ys\', \'name\', \'gate_gradients\', \'aggregation_method\', \'stop_gradients\', \'unconnect…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 1357 …_gradients_with_ops\', \'gate_gradients\', \'aggregation_method\', \'stop_gradients\', \'unconnect…
|