frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

add relevant function (#133946)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/133946
Approved by: https://github.com/ezyang
This commit is contained in:
FFFrog
2024-08-21 16:09:42 +00:00
committed by PyTorch MergeBot
parent 8604c0a150
commit a5ef04a3b8
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
torch/_C/__init__.pyi.in
View File

@ -1494,6 +1494,7 @@ class _DispatchOperatorHandle:
def debug(self) -> str: ...
class _DispatchModule:
def reset(self) -> None: ...
def def_(self, schema: str, alias: str = "") -> _DispatchModule: ...
def def_legacy(self, schema: str) -> _DispatchModule: ...
def def_name_t_t(
@ -1515,9 +1516,16 @@ class _DispatchModule:
dispatch: str,
debug: str = "impl_t_t",
) -> _DispatchModule: ...
def impl(self, name: str, dispatch: str, func: Callable) -> _DispatchModule: ...
def define(self, schema: str, alias: str = "") -> _DispatchModule: ...
def impl_with_aoti_compile(
self,
ns: str,
op_name_with_overload: str,
dispatch: _dispatchkey
) -> None: ...
def impl(self, name: str, dispatch: _dispatchkey, func: Callable) -> None: ...
def define(self, schema: str, alias: str = "") -> str: ...
def fallback_fallthrough(self, dispatch: str = "") -> _DispatchModule: ...
def fallback(self, dispatch: _dispatchkey, func: Callable, with_keyset: _bool = False) -> None: ...
_after_ADInplaceOrView_keyset: DispatchKeySet
_after_autograd_keyset: DispatchKeySet