Searched refs:forward_gradient_function (Results 1 – 1 of 1) sorted by relevance
825 PyObject* forward_gradient_function = nullptr; variable954 if (forward_gradient_function != nullptr) { in TFE_Py_RegisterJVPFunction()955 Py_DECREF(forward_gradient_function); in TFE_Py_RegisterJVPFunction()958 forward_gradient_function = nullptr; in TFE_Py_RegisterJVPFunction()965 forward_gradient_function = e; in TFE_Py_RegisterJVPFunction()2451 if (forward_gradient_function == nullptr) {2466 PyObject_CallObject(forward_gradient_function, callback_args.get()));