Searched refs:stop_gradients (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/kfac/python/ops/ |
| D | curvature_matrix_vector_products.py | 83 stop_gradients=self._wrt_tensors)
93 stop_gradients=self._wrt_tensors)
|
| D | utils.py | 298 def fwd_gradients(ys, xs, grad_xs=None, stop_gradients=None): argument
309 ys, xs, grad_ys=us, stop_gradients=stop_gradients)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_impl.py | 407 stop_gradients=None): argument
486 stop_gradients = [] if stop_gradients is None else _AsList(stop_gradients)
494 list(ys) + list(xs) + list(stop_gradients) + list(grad_ys)) as grad_scope:
517 stop_gradient_ops = [t.op for t in stop_gradients]
|
| D | gradients_test.py | 383 partialg = gradients.gradients(z, [x, y], stop_gradients=[x, y])
387 def _MakeGraph(rng, stop_gradients=()): argument
394 if "a" in stop_gradients: a = array_ops.stop_gradient(a)
396 if "b" in stop_gradients: b = array_ops.stop_gradient(b)
398 if "c" in stop_gradients: c = array_ops.stop_gradient(c)
400 if "d" in stop_gradients: d = array_ops.stop_gradient(d)
422 stop_gradients=[graph[v] for v in constants])
|
| /external/tensorflow/tensorflow/tools/api/golden/ |
| D | tensorflow.pbtxt | 1141 …_gradients_with_ops\', \'gate_gradients\', \'aggregation_method\', \'stop_gradients\'], varargs=No…
|