Home
last modified time | relevance | path

Searched defs:DispatchKey (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/pytorch/torchgen/
H A Dgen.py1297 f: NativeFunction, backend_indices: dict[DispatchKey, BackendIndex]
1433 backend_indices: dict[DispatchKey, BackendIndex],
1490 backend_indices: dict[DispatchKey, BackendIndex],
1533 dispatch_key: DispatchKey,
1635 dispatch_key: DispatchKey,
1725 backend_indices: dict[DispatchKey, BackendIndex],
1728 functions_keys: set[DispatchKey],
1729 dispatch_keys: Sequence[DispatchKey],
1845 backend_indices: dict[DispatchKey, BackendIndex],
1849 functions_keys: set[DispatchKey],
[all …]
H A Dgen_backend_stubs.py42 backend_indices: dict[DispatchKey, BackendIndex],
126 dispatch_key: DispatchKey,
250 backend_indices: dict[DispatchKey, BackendIndex],
251 backend_key: DispatchKey,
361 backend_indices: dict[DispatchKey, BackendIndex],
363 backend_dispatch_key: DispatchKey,
418 backend_indices: dict[DispatchKey, BackendIndex],
420 backend_dispatch_key: DispatchKey,
421 dispatch_key: DispatchKey,
H A Dgen_aoti_c_shim.py317 dispatch_key: DispatchKey,
318 backend_indices: dict[DispatchKey, BackendIndex],
348 dispatch_key: DispatchKey,
349 backend_indices: dict[DispatchKey, BackendIndex],
372 dispatch_key: DispatchKey,
373 backend_indices: dict[DispatchKey, BackendIndex],
428 dispatch_key: DispatchKey,
429 backend_indices: dict[DispatchKey, BackendIndex],
H A Dmodel.py79 class DispatchKey(Enum): class
305 def is_generic_dispatch_key(dk: DispatchKey) -> bool:
315 def is_cuda_dispatch_key(dk: DispatchKey) -> bool:
327 def is_xpu_dispatch_key(dk: DispatchKey) -> bool:
340 def is_structured_dispatch_key(dk: DispatchKey) -> bool:
344 def is_ufunc_dispatch_key(dk: DispatchKey) -> bool:
1284 parent_index: dict[DispatchKey, dict[OperatorName, BackendMetadata]],
1285 child_index: dict[DispatchKey, dict[OperatorName, BackendMetadata]],
H A Dcontext.py123 func: Callable[[F, dict[DispatchKey, BackendIndex]], T]
126 def wrapper(f: F, backend_indices: dict[DispatchKey, BackendIndex]) -> T:
H A Dnative_function_generation.py381 indices: dict[DispatchKey, dict[OperatorName, BackendMetadata]],
/aosp_15_r20/external/pytorch/torchgen/executorch/
H A Dmodel.py163 backend_indices: dict[DispatchKey, dict[OperatorName, BackendMetadata]],
175 backend_indices: dict[DispatchKey, dict[OperatorName, BackendMetadata]]
184 self, backend_indices: dict[DispatchKey, dict[OperatorName, BackendMetadata]]
/aosp_15_r20/external/pytorch/torchgen/api/
H A Dufunc.py28 def schema_kernel_name(func: FunctionSchema, dispatch_key: DispatchKey) -> str:
33 def kernel_name(g: NativeFunctionsGroup, dispatch_key: DispatchKey) -> str:
/aosp_15_r20/external/pytorch/torch/utils/
H A D_python_dispatch.py208 def _pop_mode(k: Optional[Union[DispatchKey, torch._C._TorchDispatchModeKey]] = None):
219 def _pop_mode_temporarily(k: Optional[DispatchKey] = None):
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/api/
H A DShaderRegistry.h23 enum class DispatchKey : int8_t { enum
/aosp_15_r20/external/executorch/backends/vulkan/runtime/api/
H A DShaderRegistry.h27 enum class DispatchKey : int8_t { enum
/aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/
H A Dop_allowlist.h188 constexpr bool dispatch_key_allowlist_check(DispatchKey /*k*/) { in dispatch_key_allowlist_check() argument
/aosp_15_r20/external/pytorch/torch/
H A D_ops.py177 def resolve_key(op: OperatorBase, k: DispatchKey): # type: ignore[valid-type]
904 def _may_use_fallthrough_instead_of_fallback(key: DispatchKey):
H A Dtypes.py21 DispatchKey as DispatchKey, unknown
/aosp_15_r20/external/pytorch/c10/core/
H A DDispatchKey.h135 enum class DispatchKey : uint16_t { enum
/aosp_15_r20/external/pytorch/torch/_refs/nn/functional/
H A D__init__.py72 DispatchKey = torch._C.DispatchKey # type: ignore[attr-defined] variable
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py354 DispatchKey = torch._C.DispatchKey # type: ignore[attr-defined] variable
/aosp_15_r20/external/pytorch/torch/_decomp/
H A Ddecompositions.py38 DispatchKey = torch._C.DispatchKey # type: ignore[attr-defined] variable