| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | gen.py | 323 def static_dispatch_keys(backends: list[BackendIndex]) -> list[DispatchKey]:
336 f: NativeFunction, backend_index: BackendIndex
356 f: NativeFunction, backend_index: list[BackendIndex]
371 def static_dispatch_extra_headers(backends: list[BackendIndex]) -> list[str]:
419 backend_index: BackendIndex,
437 backend_indices: list[BackendIndex],
466 backend_indices: list[BackendIndex],
1297 f: NativeFunction, backend_indices: dict[DispatchKey, BackendIndex] argument
1433 backend_indices: dict[DispatchKey, BackendIndex], argument
1490 backend_indices: dict[DispatchKey, BackendIndex], argument
[all …]
|
| H A D | gen_aoti_c_shim.py | 307 backend_index: BackendIndex,
318 backend_indices: dict[DispatchKey, BackendIndex], argument
349 backend_indices: dict[DispatchKey, BackendIndex], argument
373 backend_indices: dict[DispatchKey, BackendIndex], argument
429 backend_indices: dict[DispatchKey, BackendIndex], argument
|
| H A D | context.py | 111 func: Callable[[F, BackendIndex], T] argument
114 def wrapper(f: F, backend_index: BackendIndex) -> T:
123 func: Callable[[F, dict[DispatchKey, BackendIndex]], T] argument
126 def wrapper(f: F, backend_indices: dict[DispatchKey, BackendIndex]) -> T: argument
|
| H A D | gen_backend_stubs.py | 42 backend_indices: dict[DispatchKey, BackendIndex], argument
250 backend_indices: dict[DispatchKey, BackendIndex], argument
361 backend_indices: dict[DispatchKey, BackendIndex], argument
418 backend_indices: dict[DispatchKey, BackendIndex], argument
|
| H A D | gen_functionalization_type.py | 777 composite_implicit_autograd_index: BackendIndex,
|
| H A D | gen_executorch.py | 82 backend_indices: list[BackendIndex],
|
| H A D | model.py | 1268 class BackendIndex: class
2644 f: NativeFunction, g: NativeFunctionsGroup, b: BackendIndex
|
| /aosp_15_r20/external/pytorch/torchgen/static_runtime/ |
| H A D | generator.py | 511 def get_kernel_name(g: NativeFunctionsGroup, backend_index: BackendIndex) -> str:
518 def get_out_kernel_name(g: NativeFunctionsGroup, backend_index: BackendIndex) -> str:
526 g: NativeFunctionsGroup, backend_index: BackendIndex
537 g: NativeFunctionsViewGroup, backend_index: BackendIndex
550 g: NativeFunctionsGroup, backend_index: BackendIndex
605 self, groups: Sequence[NativeFunctionsGroup], backend_index: BackendIndex
631 self, groups: Sequence[NativeFunctionsViewGroup], backend_index: BackendIndex
657 self, g: NativeFunctionsGroup, backend_index: BackendIndex
682 self, g: NativeFunctionsViewGroup, backend_index: BackendIndex
|
| /aosp_15_r20/external/pytorch/torchgen/dest/ |
| H A D | register_dispatch_key.py | 47 backend_index: BackendIndex,
83 backend_index: BackendIndex,
110 def gen_create_out_helper(backend_index: BackendIndex) -> list[str]:
133 def gen_maybe_create_proxy_helper(backend_index: BackendIndex) -> list[str]:
151 def gen_resize_out_helper(backend_index: BackendIndex) -> list[str]:
181 def gen_check_inplace_helper(backend_index: BackendIndex) -> list[str]:
204 def gen_registration_helpers(backend_index: BackendIndex) -> list[str]:
|
| H A D | native_functions.py | 12 def gen_unstructured(f: NativeFunction, backend_index: BackendIndex) -> str | None:
25 def gen_structured(g: NativeFunctionsGroup, backend_index: BackendIndex) -> list[str]:
45 g: NativeFunctionsGroup | NativeFunction, backend_index: BackendIndex
|
| /aosp_15_r20/external/pytorch/torchgen/api/types/ |
| H A D | signatures.py | 395 f: NativeFunction, backend_index: BackendIndex, *, prefix: str = ""
|