This doesn’t seem to be an autograd.Function.
You could run the code line by line in a debugger (or notebook etc.) in order to isolate the line of code which raises the error, if you cannot find the definition (it could be defined in another submodule you are importing).