MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / TFE_Py_RecordGradient

Function TFE_Py_RecordGradient

tensorflow/python/eager/pywrap_tfe_src.cc:3233–3241  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

3231}
3232
3233PyObject* TFE_Py_RecordGradient(PyObject* op_name, PyObject* inputs,
3234 PyObject* attrs, PyObject* results,
3235 PyObject* name) {
3236 if (*ThreadTapeIsStopped() || !HasTape()) {
3237 Py_RETURN_NONE;
3238 }
3239
3240 return RecordGradient(op_name, inputs, attrs, results, name);
3241}
3242
3243namespace {
3244const char kTensor[] = "T";

Callers

nothing calls this directly

Calls 2

ThreadTapeIsStoppedFunction · 0.85
RecordGradientFunction · 0.85

Tested by

no test coverage detected