| /external/tensorflow/tensorflow/python/eager/ |
| D | imperative_grad.py | 24 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
38 unconnected_gradients=UnconnectedGradients.NONE): argument
66 unconnected_gradients = UnconnectedGradients(unconnected_gradients)
69 "Unknown value for unconnected_gradients: %r" % unconnected_gradients)
77 compat.as_str(unconnected_gradients.value))
|
| D | forwardprop.py | 36 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
136 unconnected_gradients=UnconnectedGradients.ZERO)
417 def jvp(self, primals, unconnected_gradients=UnconnectedGradients.NONE): argument
435 unconnected_gradients = UnconnectedGradients(unconnected_gradients)
446 if result is None and unconnected_gradients == UnconnectedGradients.ZERO:
|
| D | backprop.py | 49 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
991 unconnected_gradients=UnconnectedGradients.NONE): argument
1080 unconnected_gradients=unconnected_gradients)
1093 unconnected_gradients=UnconnectedGradients.NONE, argument
1166 unconnected_gradients=unconnected_gradients)
1209 unconnected_gradients=UnconnectedGradients.NONE, argument
1307 unconnected_gradients=unconnected_gradients)
|
| D | forwardprop_test.py | 49 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
64 primals_out, unconnected_gradients=UnconnectedGradients.ZERO)
120 unconnected_gradients=UnconnectedGradients.ZERO)
353 self.assertAllClose([[0.]], acc.jvp(v, unconnected_gradients="zero"))
849 self.assertAllClose(0.1, acc.jvp(x, unconnected_gradients="zero"))
850 self.assertAllClose(0.1, acc.jvp(x, unconnected_gradients="none"))
852 self.assertAllClose(0.0, acc.jvp(y, unconnected_gradients="zero"))
853 self.assertIsNone(acc.jvp(y, unconnected_gradients="none"))
|
| D | pywrap_tfe.h | 256 PyObject* unconnected_gradients,
|
| D | backprop_test.py | 981 dz_dx = g.gradient(z, x, unconnected_gradients='zero')
995 dz_dx = g.gradient(z, x, unconnected_gradients='zero')
1007 g.gradient(z, x, unconnected_gradients='nonsense')
1027 dy_dx = g.gradient(y, x, unconnected_gradients='zero')
2002 jac = tape.batch_jacobian(y, x, unconnected_gradients='zero',
|
| D | BUILD | 653 "//tensorflow/python:unconnected_gradients",
818 "//tensorflow/python:unconnected_gradients",
|
| D | pywrap_tfe_src.cc | 2721 PyObject* unconnected_gradients, argument
2796 strcmp(TFE_GetPythonString(unconnected_gradients), "zero") == 0;
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scaling_gradient_tape.py | 27 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
131 unconnected_gradients=UnconnectedGradients.NONE): argument
177 unconnected_gradients))
186 unconnected_gradients=UnconnectedGradients.NONE, argument
196 unconnected_gradients=UnconnectedGradients.NONE, argument
206 output_gradients, unconnected_gradients): argument
282 unconnected_gradients)
|
| D | BUILD | 124 "//tensorflow/python:unconnected_gradients",
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_impl.py | 39 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
52 unconnected_gradients=UnconnectedGradients.NONE): argument
172 unconnected_gradients)
184 unconnected_gradients=UnconnectedGradients.NONE): argument
318 unconnected_gradients)
|
| D | gradients_util.py | 45 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
488 unconnected_gradients=UnconnectedGradients.NONE, argument
497 unconnected_gradients = UnconnectedGradients(unconnected_gradients)
500 "Unknown value for unconnected_gradients: %r" % unconnected_gradients)
731 return [_GetGrad(grads, x, unconnected_gradients) for x in xs]
808 def _GetGrad(grads, t, unconnected_gradients): argument
813 if unconnected_gradients == UnconnectedGradients.ZERO:
815 elif unconnected_gradients == UnconnectedGradients.NONE:
819 "Unknown value for unconnected_gradients: %r" % unconnected_gradients)
823 if unconnected_gradients == UnconnectedGradients.ZERO and t_grad is None:
|
| D | gradients_test.py | 58 from tensorflow.python.ops import unconnected_gradients
304 unconnected_gradients=unconnected_gradients.UnconnectedGradients.ZERO
405 [y], [x], unconnected_gradients="none")
413 [y], [x], unconnected_gradients="zero")
422 [y], [x], unconnected_gradients="zero")
432 gradients.gradients([y], [x], unconnected_gradients="nonsense")
434 @parameterized.parameters(unconnected_gradients.UnconnectedGradients.ZERO,
435 unconnected_gradients.UnconnectedGradients.NONE)
454 e, d, unconnected_gradients=unconnected_gradients_val)
456 unconnected_gradients.UnconnectedGradients.ZERO):
[all …]
|
| D | gradients.py | 29 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
|
| D | custom_gradient.py | 31 from tensorflow.python.ops.unconnected_gradients import UnconnectedGradients
541 unconnected_gradients=UnconnectedGradients.ZERO)
|
| D | while_v2.py | 706 unconnected_gradients="zero")
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| 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.-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 | 701 …\'gate_gradients\', \'aggregation_method\', \'stop_gradients\', \'unconnected_gradients\'], vararg…
|
| /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…
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | forwardprop_test.py | 32 primals_out, unconnected_gradients=tf.UnconnectedGradients.ZERO)
67 unconnected_gradients=tf.UnconnectedGradients.ZERO)
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 933 const py::handle& unconnected_gradients) { in PYBIND11_MODULE() argument
938 sources_raw.ptr(), unconnected_gradients.ptr(), status.get()); in PYBIND11_MODULE()
|
| D | BUILD | 3438 ":unconnected_gradients",
3474 ":unconnected_gradients",
3500 ":unconnected_gradients",
3543 name = "unconnected_gradients",
3544 srcs = ["ops/unconnected_gradients.py"],
4697 ":unconnected_gradients",
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | resource_variable_ops_test.py | 340 g = gradients_impl.gradients(c, [b], unconnected_gradients="zero")[0]
|