Searched refs:gradient_function (Results 1 – 1 of 1) sorted by relevance
705 PyObject* gradient_function = nullptr; variable799 if (gradient_function != nullptr) { in TFE_Py_RegisterGradientFunction()800 Py_DECREF(gradient_function); in TFE_Py_RegisterGradientFunction()803 gradient_function = nullptr; in TFE_Py_RegisterGradientFunction()810 gradient_function = e; in TFE_Py_RegisterGradientFunction()2145 PyObject_CallObject(gradient_function, callback_args.get()));