Searched refs:torch_func_api (Results 1 – 1 of 1) sorted by relevance
44 def setup_docs(functorch_api, torch_func_api=None, new_api_name=None): argument46 if torch_func_api is None:47 torch_func_api = getattr(_impl, api_name)49 if torch_func_api.__doc__ is None:55 functorch_api.__doc__ = torch_func_api.__doc__ + warning_note