Home
last modified time | relevance | path

Searched refs:NativeFunctionsGroup (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/pytorch/torchgen/static_runtime/
H A Dgenerator.py16 NativeFunctionsGroup,
242 def is_supported(g: NativeFunctionsGroup | NativeFunctionsViewGroup) -> bool:
511 def get_kernel_name(g: NativeFunctionsGroup, backend_index: BackendIndex) -> str: argument
518 def get_out_kernel_name(g: NativeFunctionsGroup, backend_index: BackendIndex) -> str: argument
526 g: NativeFunctionsGroup, backend_index: BackendIndex argument
550 g: NativeFunctionsGroup, backend_index: BackendIndex argument
599 def op_name_from_group(g: NativeFunctionsGroup) -> str: argument
605 self, groups: Sequence[NativeFunctionsGroup], backend_index: BackendIndex argument
613 assert isinstance(g, NativeFunctionsGroup)
657 self, g: NativeFunctionsGroup, backend_index: BackendIndex argument
[all …]
H A Dgen_static_runtime_ops.py12 from torchgen.model import DispatchKey, NativeFunctionsGroup, NativeFunctionsViewGroup
22 bound=Union[NativeFunctionsGroup, NativeFunctionsViewGroup],
33 def is_supported(g: NativeFunctionsGroup | NativeFunctionsViewGroup) -> bool:
169 if isinstance(g, NativeFunctionsGroup)
H A Dconfig.py3 from torchgen.model import NativeFunctionsGroup, NativeFunctionsViewGroup
6 def func_name_base_str(g: NativeFunctionsGroup | NativeFunctionsViewGroup) -> str:
7 if isinstance(g, NativeFunctionsGroup):
58 def is_hand_written(g: NativeFunctionsGroup | NativeFunctionsViewGroup) -> bool:
/aosp_15_r20/external/pytorch/torchgen/dest/
H A Dufunc.py25 NativeFunctionsGroup,
67 g: NativeFunctionsGroup
102 g: NativeFunctionsGroup
131 def eligible_for_binary_scalar_specialization(g: NativeFunctionsGroup) -> bool: argument
139 g: NativeFunctionsGroup, argument
243 g: NativeFunctionsGroup, argument
290 def compute_ufunc_cuda(g: NativeFunctionsGroup) -> str: argument
340 g: NativeFunctionsGroup
383 def compute_ufunc_cpu(g: NativeFunctionsGroup) -> str: argument
399 g: NativeFunctionsGroup, argument
[all …]
H A Dnative_functions.py7 from torchgen.model import BackendIndex, NativeFunction, NativeFunctionsGroup
25 def gen_structured(g: NativeFunctionsGroup, backend_index: BackendIndex) -> list[str]: argument
45 g: NativeFunctionsGroup | NativeFunction, backend_index: BackendIndex
48 if isinstance(g, NativeFunctionsGroup):
H A Dregister_dispatch_key.py35 NativeFunctionsGroup,
285 def __call__(self, f: NativeFunctionsGroup | NativeFunction) -> list[str]:
286 if isinstance(f, NativeFunctionsGroup):
287 g: NativeFunctionsGroup = f
313 self, f: NativeFunction, g: NativeFunctionsGroup | None
363 def gen_structured(self, g: NativeFunctionsGroup) -> list[str]: argument
393 self, f: NativeFunction, g: NativeFunctionsGroup | None = None
581 g: NativeFunctionsGroup
H A Dlazy_ir.py39 NativeFunctionsGroup,
176 def __call__(self, f: NativeFunctionsGroup | NativeFunction) -> list[str]:
177 func = f.functional.func if isinstance(f, NativeFunctionsGroup) else f.func
179 f.functional if isinstance(f, NativeFunctionsGroup) else f
/aosp_15_r20/external/pytorch/torchgen/
H A Dcontext.py12 NativeFunctionsGroup,
23 NativeFunctionsGroup,
25 Union[NativeFunction, NativeFunctionsGroup],
32 NativeFunctionsGroup,
43 g: NativeFunctionsGroup | NativeFunctionsViewGroup | NativeFunction,
45 if isinstance(g, NativeFunctionsGroup):
H A Dgen.py65 NativeFunctionsGroup,
792 def compute_meta_function_declaration(g: NativeFunctionsGroup) -> str | None: argument
1410 ) -> Sequence[NativeFunction | NativeFunctionsGroup]:
1413 ) -> Sequence[NativeFunction | NativeFunctionsGroup]:
1414 r = NativeFunctionsGroup.from_dict(d)
1432 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup], argument
1435 [NativeFunctionsGroup | NativeFunction, BackendIndex], list[str]
1489 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup], argument
1492 [NativeFunctionsGroup | NativeFunction, BackendIndex], list[str]
1514 *, f: NativeFunction | NativeFunctionsGroup, backend_idx: BackendIndex argument
[all …]
H A Dgen_lazy_tensor.py22 from torchgen.model import NativeFunction, NativeFunctionsGroup, OperatorName
96 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup], argument
341 def sort_native_function(f: NativeFunctionsGroup | NativeFunction) -> str:
346 func = f.functional.func if isinstance(f, NativeFunctionsGroup) else f.func
371 xs: Iterable[NativeFunctionsGroup | NativeFunction],
380 fs = list(x.functions()) if isinstance(x, NativeFunctionsGroup) else [x]
H A Dgen_aoti_c_shim.py19 NativeFunctionsGroup,
316 func_group_mapping: dict[OperatorName, NativeFunctionsGroup], argument
347 func_group_mapping: dict[OperatorName, NativeFunctionsGroup], argument
371 func_group_mapping: dict[OperatorName, NativeFunctionsGroup], argument
405 func_group_mapping: dict[OperatorName, NativeFunctionsGroup]
427 func_group_mapping: dict[OperatorName, NativeFunctionsGroup], argument
H A Dgen_functionalization_type.py39 NativeFunctionsGroup,
617 f: NativeFunction, g: NativeFunctionsGroup argument
776 g: NativeFunction | NativeFunctionsGroup | NativeFunctionsViewGroup,
805 elif isinstance(g, NativeFunctionsGroup):
842 g: NativeFunction | NativeFunctionsGroup | NativeFunctionsViewGroup,
H A Dgen_backend_stubs.py23 NativeFunctionsGroup,
41 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup], argument
362 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup], argument
419 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup], argument
H A Dnative_function_generation.py23 NativeFunctionsGroup,
529 def gen_composite_functional_kernel(g: NativeFunctionsGroup) -> str | None: argument
592 def gen_composite_out_kernel(g: NativeFunctionsGroup) -> str | None: argument
H A Dmodel.py1100 class NativeFunctionsGroup: class
1192 def from_dict(d: dict[SchemaKind, NativeFunction]) -> NativeFunctionsGroup | None:
1208 return NativeFunctionsGroup(
1294 def primary(self, g: NativeFunctionsGroup) -> NativeFunction: argument
1300 def has_kernel(self, g: NativeFunction | NativeFunctionsGroup) -> bool: argument
1305 self, g: NativeFunction | NativeFunctionsGroup argument
1309 elif isinstance(g, NativeFunctionsGroup):
2644 f: NativeFunction, g: NativeFunctionsGroup, b: BackendIndex argument
H A Dgen_executorch.py46 NativeFunctionsGroup,
333 g: NativeFunctionsGroup | NativeFunction, kernel_index: ETKernelIndex
413 NativeFunctionsGroup | NativeFunction,
/aosp_15_r20/external/pytorch/torchgen/api/
H A Dufunc.py23 NativeFunctionsGroup,
33 def kernel_name(g: NativeFunctionsGroup, dispatch_key: DispatchKey) -> str: argument
160 g: NativeFunctionsGroup, *, scalar_tensor_idx: int | None, scalar_t: BaseCppType argument
189 def ufunc_arguments(g: NativeFunctionsGroup, *, compute_t: CType) -> list[Binding]: argument
201 def stub_arguments(g: NativeFunctionsGroup) -> list[Binding]: argument
H A Dstructured.py27 NativeFunctionsGroup,
118 def impl_arguments(g: NativeFunctionsGroup) -> list[Binding]: argument
148 def meta_arguments(g: NativeFunctionsGroup) -> list[Binding]: argument
154 def out_arguments(g: NativeFunctionsGroup) -> list[Binding]: argument
H A Dmeta.py1 from torchgen.model import NativeFunctionsGroup
11 def name(g: NativeFunctionsGroup) -> str: argument
/aosp_15_r20/external/pytorch/torchgen/executorch/
H A Dmodel.py16 NativeFunctionsGroup,
143 def has_kernels(self, g: NativeFunction | NativeFunctionsGroup) -> bool: argument
148 self, g: NativeFunction | NativeFunctionsGroup argument
152 elif isinstance(g, NativeFunctionsGroup):
/aosp_15_r20/external/pytorch/tools/test/
H A Dtest_codegen_model.py17 NativeFunctionsGroup,
45 assert isinstance(g, NativeFunctionsGroup)
/aosp_15_r20/external/pytorch/tools/jit/
H A Dgen_unboxing.py20 from torchgen.model import Argument, NativeFunction, NativeFunctionsGroup, Variant
165 def key_func(fn: NativeFunction | NativeFunctionsGroup) -> str: argument
/aosp_15_r20/external/pytorch/torchgen/api/types/
H A Dsignatures.py14 NativeFunctionsGroup,
380 g: NativeFunctionsGroup