Home
last modified time | relevance | path

Searched defs:NativeFunctionsGroup (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/pytorch/torchgen/static_runtime/
H A Dgenerator.py511 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
550 g: NativeFunctionsGroup, backend_index: BackendIndex
599 def op_name_from_group(g: NativeFunctionsGroup) -> str:
605 self, groups: Sequence[NativeFunctionsGroup], backend_index: BackendIndex
657 self, g: NativeFunctionsGroup, backend_index: BackendIndex
698 def out_variant(self, groups: Sequence[NativeFunctionsGroup]) -> str:
722 def out_variant_op_test_case_generator(self, g: NativeFunctionsGroup) -> str:
/aosp_15_r20/external/pytorch/torchgen/
H A Dgen.py792 def compute_meta_function_declaration(g: NativeFunctionsGroup) -> str | None:
1432 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup],
1489 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup],
1514 *, f: NativeFunction | NativeFunctionsGroup, backend_idx: BackendIndex
1532 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup],
1634 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup],
1721 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup],
1722 structured_native_functions: Sequence[NativeFunctionsGroup],
1842 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup],
2037 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup],
[all …]
H A Dgen_aoti_c_shim.py316 func_group_mapping: dict[OperatorName, NativeFunctionsGroup], argument
347 func_group_mapping: dict[OperatorName, NativeFunctionsGroup], argument
371 func_group_mapping: dict[OperatorName, NativeFunctionsGroup], argument
427 func_group_mapping: dict[OperatorName, NativeFunctionsGroup], argument
H A Dgen_backend_stubs.py41 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup],
362 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup],
419 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup],
H A Dnative_function_generation.py529 def gen_composite_functional_kernel(g: NativeFunctionsGroup) -> str | None:
592 def gen_composite_out_kernel(g: NativeFunctionsGroup) -> str | None:
H A Dmodel.py1100 class NativeFunctionsGroup: class
1294 def primary(self, g: NativeFunctionsGroup) -> NativeFunction:
1300 def has_kernel(self, g: NativeFunction | NativeFunctionsGroup) -> bool:
1305 self, g: NativeFunction | NativeFunctionsGroup
2644 f: NativeFunction, g: NativeFunctionsGroup, b: BackendIndex
H A Dgen_lazy_tensor.py96 grouped_native_functions: Sequence[NativeFunction | NativeFunctionsGroup],
H A Dgen_functionalization_type.py617 f: NativeFunction, g: NativeFunctionsGroup
/aosp_15_r20/external/pytorch/torchgen/dest/
H A Dufunc.py131 def eligible_for_binary_scalar_specialization(g: NativeFunctionsGroup) -> bool:
139 g: NativeFunctionsGroup,
243 g: NativeFunctionsGroup,
290 def compute_ufunc_cuda(g: NativeFunctionsGroup) -> str:
383 def compute_ufunc_cpu(g: NativeFunctionsGroup) -> str:
399 g: NativeFunctionsGroup,
492 def compute_ufunc_cpu_kernel(g: NativeFunctionsGroup) -> str:
H A Dnative_functions.py25 def gen_structured(g: NativeFunctionsGroup, backend_index: BackendIndex) -> list[str]:
H A Dregister_dispatch_key.py363 def gen_structured(self, g: NativeFunctionsGroup) -> list[str]:
/aosp_15_r20/external/pytorch/torchgen/api/
H A Dufunc.py33 def kernel_name(g: NativeFunctionsGroup, dispatch_key: DispatchKey) -> str:
160 g: NativeFunctionsGroup, *, scalar_tensor_idx: int | None, scalar_t: BaseCppType
189 def ufunc_arguments(g: NativeFunctionsGroup, *, compute_t: CType) -> list[Binding]:
201 def stub_arguments(g: NativeFunctionsGroup) -> list[Binding]:
H A Dstructured.py118 def impl_arguments(g: NativeFunctionsGroup) -> list[Binding]:
148 def meta_arguments(g: NativeFunctionsGroup) -> list[Binding]:
154 def out_arguments(g: NativeFunctionsGroup) -> list[Binding]:
H A Dmeta.py11 def name(g: NativeFunctionsGroup) -> str:
/aosp_15_r20/external/pytorch/torchgen/executorch/
H A Dmodel.py143 def has_kernels(self, g: NativeFunction | NativeFunctionsGroup) -> bool:
148 self, g: NativeFunction | NativeFunctionsGroup
/aosp_15_r20/external/pytorch/tools/jit/
H A Dgen_unboxing.py165 def key_func(fn: NativeFunction | NativeFunctionsGroup) -> str: