| /external/tensorflow/tensorflow/python/eager/ |
| D | imperative_grad.py | 20 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
34 unconnected_gradients=UnconnectedGradients.NONE): argument
62 unconnected_gradients = UnconnectedGradients(unconnected_gradients)
65 "Unknown value for unconnected_gradients: %r" % unconnected_gradients)
73 compat.as_str(unconnected_gradients.value))
|
| D | forwardprop.py | 32 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
132 unconnected_gradients=UnconnectedGradients.ZERO)
413 def jvp(self, primals, unconnected_gradients=UnconnectedGradients.NONE): argument
431 unconnected_gradients = UnconnectedGradients(unconnected_gradients)
442 if result is None and unconnected_gradients == UnconnectedGradients.ZERO:
|
| D | backprop.py | 46 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
1011 unconnected_gradients=UnconnectedGradients.NONE): argument
1118 unconnected_gradients=unconnected_gradients)
1134 unconnected_gradients=UnconnectedGradients.NONE, argument
1207 unconnected_gradients=unconnected_gradients)
1247 unconnected_gradients=UnconnectedGradients.NONE, argument
1359 unconnected_gradients=unconnected_gradients)
|
| D | forwardprop_test.py | 46 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
61 primals_out, unconnected_gradients=UnconnectedGradients.ZERO)
121 unconnected_gradients=UnconnectedGradients.ZERO)
357 self.assertAllClose([[0.]], acc.jvp(v, unconnected_gradients="zero"))
893 self.assertAllClose(0.1, acc.jvp(x, unconnected_gradients="zero"))
894 self.assertAllClose(0.1, acc.jvp(x, unconnected_gradients="none"))
896 self.assertAllClose(0.0, acc.jvp(y, unconnected_gradients="zero"))
897 self.assertIsNone(acc.jvp(y, unconnected_gradients="none"))
|
| D | pywrap_tfe.h | 260 PyObject* unconnected_gradients,
|
| D | backprop_test.py | 1030 dz_dx = g.gradient(z, x, unconnected_gradients='zero')
1044 dz_dx = g.gradient(z, x, unconnected_gradients='zero')
1056 g.gradient(z, x, unconnected_gradients='nonsense')
1076 dy_dx = g.gradient(y, x, unconnected_gradients='zero')
2059 jac = tape.batch_jacobian(y, x, unconnected_gradients='zero',
|
| D | BUILD | 709 "//tensorflow/python:unconnected_gradients",
917 "//tensorflow/python:unconnected_gradients",
|
| D | pywrap_tfe_src.cc | 2799 PyObject* unconnected_gradients, argument
2873 strcmp(TFE_GetPythonString(unconnected_gradients), "zero") == 0;
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_impl.py | 35 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
48 unconnected_gradients=UnconnectedGradients.NONE): argument
168 unconnected_gradients)
180 unconnected_gradients=UnconnectedGradients.NONE): argument
314 unconnected_gradients)
|
| D | gradients_util.py | 41 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
476 unconnected_gradients=UnconnectedGradients.NONE, argument
501 unconnected_gradients, src_graph)
508 unconnected_gradients = UnconnectedGradients(unconnected_gradients)
742 return [_GetGrad(grads, x, unconnected_gradients) for x in xs]
819 def _GetGrad(grads, t, unconnected_gradients): argument
824 if unconnected_gradients == UnconnectedGradients.ZERO:
826 elif unconnected_gradients == UnconnectedGradients.NONE:
834 if unconnected_gradients == UnconnectedGradients.ZERO and t_grad is None:
|
| D | gradients_test.py | 57 from tensorflow.python.ops import unconnected_gradients
303 unconnected_gradients=unconnected_gradients.UnconnectedGradients.ZERO
404 [y], [x], unconnected_gradients="none")
412 [y], [x], unconnected_gradients="zero")
421 [y], [x], unconnected_gradients="zero")
431 gradients.gradients([y], [x], unconnected_gradients="nonsense")
433 @parameterized.parameters(unconnected_gradients.UnconnectedGradients.ZERO,
434 unconnected_gradients.UnconnectedGradients.NONE)
453 e, d, unconnected_gradients=unconnected_gradients_val)
455 unconnected_gradients.UnconnectedGradients.ZERO):
[all …]
|
| D | gradients.py | 25 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
|
| D | custom_gradient.py | 29 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
734 unconnected_gradients=UnconnectedGradients.ZERO)
|
| D | while_v2.py | 755 unconnected_gradients="zero")
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-gradient-tape.pbtxt | 11 …argspec: "args=[\'self\', \'target\', \'source\', \'unconnected_gradients\', \'parallel_iterations…
15 …"args=[\'self\', \'target\', \'sources\', \'output_gradients\', \'unconnected_gradients\'], vararg…
19 …argspec: "args=[\'self\', \'target\', \'sources\', \'unconnected_gradients\', \'parallel_iteration…
|
| D | tensorflow.pbtxt | 1429 …\'gate_gradients\', \'aggregation_method\', \'stop_gradients\', \'unconnected_gradients\'], vararg…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.autodiff.-gradient-tape.pbtxt | 11 …argspec: "args=[\'self\', \'target\', \'source\', \'unconnected_gradients\', \'parallel_iterations…
15 …"args=[\'self\', \'target\', \'sources\', \'output_gradients\', \'unconnected_gradients\'], vararg…
19 …argspec: "args=[\'self\', \'target\', \'sources\', \'unconnected_gradients\', \'parallel_iteration…
|
| D | tensorflow.-gradient-tape.pbtxt | 11 …argspec: "args=[\'self\', \'target\', \'source\', \'unconnected_gradients\', \'parallel_iterations…
15 …"args=[\'self\', \'target\', \'sources\', \'output_gradients\', \'unconnected_gradients\'], vararg…
19 …argspec: "args=[\'self\', \'target\', \'sources\', \'unconnected_gradients\', \'parallel_iteration…
|
| D | tensorflow.autodiff.-forward-accumulator.pbtxt | 11 …argspec: "args=[\'self\', \'primals\', \'unconnected_gradients\'], varargs=None, keywords=None, de…
|
| D | tensorflow.pbtxt | 709 …\'gate_gradients\', \'aggregation_method\', \'stop_gradients\', \'unconnected_gradients\'], vararg…
|
| /external/tensorflow/tensorflow/python/ |
| D | BUILD | 1572 ":unconnected_gradients",
1606 ":unconnected_gradients",
1631 ":unconnected_gradients",
1677 name = "unconnected_gradients",
1678 srcs = ["ops/unconnected_gradients.py"],
2838 ":unconnected_gradients",
|
| D | tfe_wrapper.cc | 1092 const py::handle& unconnected_gradients) { in PYBIND11_MODULE() argument
1097 sources_raw.ptr(), unconnected_gradients.ptr(), status.get()); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| D | resource_variable_ops_test.py | 367 g = gradients_impl.gradients(c, [b], unconnected_gradients="zero")[0]
|