Searched defs:unconnected_gradients (Results 1 – 7 of 7) sorted by relevance
48 unconnected_gradients=UnconnectedGradients.NONE): argument180 unconnected_gradients=UnconnectedGradients.NONE): argument
476 unconnected_gradients=UnconnectedGradients.NONE, argument819 def _GetGrad(grads, t, unconnected_gradients): argument
1011 unconnected_gradients=UnconnectedGradients.NONE): argument1134 unconnected_gradients=UnconnectedGradients.NONE, argument1247 unconnected_gradients=UnconnectedGradients.NONE, argument
34 unconnected_gradients=UnconnectedGradients.NONE): argument
413 def jvp(self, primals, unconnected_gradients=UnconnectedGradients.NONE): argument
2799 PyObject* unconnected_gradients,
1092 const py::handle& unconnected_gradients) { in PYBIND11_MODULE()