Merge pull request #21055 from shoyer:relu-doc

PiperOrigin-RevId: 630425740
This commit is contained in:
jax authors 2024-05-03 10:18:49 -07:00
commit e70191bd9e

View File

@ -322,7 +322,7 @@ def linkcode_resolve(domain, info):
obj = operator.attrgetter(info['fullname'])(mod)
if isinstance(obj, property):
obj = obj.fget
if hasattr(obj, '__wrapped__'): # jit decorated functions
while hasattr(obj, '__wrapped__'): # decorated functions
obj = obj.__wrapped__
filename = inspect.getsourcefile(obj)
source, linenum = inspect.getsourcelines(obj)