Searched refs:forward_gradient_function (Results 1 – 1 of 1) sorted by relevance
836 PyObject* forward_gradient_function = nullptr; variable975 if (forward_gradient_function != nullptr) { in TFE_Py_RegisterJVPFunction()976 Py_DECREF(forward_gradient_function); in TFE_Py_RegisterJVPFunction()979 forward_gradient_function = nullptr; in TFE_Py_RegisterJVPFunction()986 forward_gradient_function = e; in TFE_Py_RegisterJVPFunction()2523 if (forward_gradient_function == nullptr) {2538 PyObject_CallObject(forward_gradient_function, callback_args.get()));