Home
last modified time | relevance | path

Searched defs:NativeFunction (Results 1 – 25 of 45) sorted by relevance

12

/aosp_15_r20/external/pytorch/torchgen/
H A Dgen.py254 def error_check_native_functions(funcs: Sequence[NativeFunction]) -> None:
336 f: NativeFunction, backend_index: BackendIndex
356 f: NativeFunction, backend_index: list[BackendIndex]
418 f: NativeFunction,
436 f: NativeFunction,
465 f: NativeFunction,
545 def __call__(self, f: NativeFunction) -> str | None:
570 def __call__(self, f: NativeFunction) -> str:
664 def __call__(self, f: NativeFunction) -> str | None:
715 def __call__(self, f: NativeFunction) -> str | None:
[all …]
H A Dgen_executorch.py81 f: NativeFunction,
130 def __call__(self, f: NativeFunction) -> str | None:
187 unbox_kernel_entry: tuple[NativeFunction, tuple[ETKernelKey, BackendMetadata]],
289 native_functions: Sequence[NativeFunction],
298 item: tuple[NativeFunction, tuple[ETKernelKey, BackendMetadata]]
333 g: NativeFunctionsGroup | NativeFunction, kernel_index: ETKernelIndex
355 native_functions: Sequence[NativeFunction],
409 native_functions: Sequence[NativeFunction],
413 NativeFunctionsGroup | NativeFunction,
440 native_functions: Sequence[NativeFunction],
[all …]
H A Dgen_functionalization_type.py178 def modifies_arguments(f: NativeFunction) -> bool:
517 def maybe_create_output(f: NativeFunction, var_name: str) -> str:
529 f: NativeFunction, inner_return_var: str
549 f: NativeFunction, inner_return_var: str
557 f: NativeFunction, functional_op: NativeFunction, inner_return_var: str
617 f: NativeFunction, g: NativeFunctionsGroup
780 def emit_registration_helper(f: NativeFunction) -> str:
H A Dgen_aoti_c_shim.py291 f: NativeFunction,
306 f: NativeFunction,
315 func: NativeFunction,
346 func: NativeFunction,
361 def get_fallback_op_name(func: NativeFunction) -> str:
370 func: NativeFunction,
413 func: NativeFunction,
426 native_functions: Sequence[NativeFunction],
H A Dgen_vmap_plumbing.py135 def gen_vmap_inplace_plumbing(native_function: NativeFunction) -> str | None:
179 def gen_vmap_plumbing_no_returns(native_function: NativeFunction) -> str:
200 def gen_vmap_plumbing(native_function: NativeFunction) -> str | None:
254 def __call__(self, f: NativeFunction) -> str | None:
259 def gen_all_vmap_plumbing(native_functions: Sequence[NativeFunction]) -> str:
H A Dnative_function_generation.py105 native_functions: Sequence[NativeFunction],
270 f: NativeFunction, k: SchemaKind
380 rs: list[NativeFunction],
H A Dgen_lazy_tensor.py341 def sort_native_function(f: NativeFunctionsGroup | NativeFunction) -> str:
370 func: Callable[[NativeFunction], Sequence[str]],
371 xs: Iterable[NativeFunctionsGroup | NativeFunction],
/aosp_15_r20/external/pytorch/tools/autograd/
H A Dgen_python_functions.py193 def should_generate_py_binding(f: NativeFunction) -> bool:
222 def is_py_variable_method(f: NativeFunction) -> bool:
226 def is_py_torch_function(f: NativeFunction) -> bool:
230 def is_py_nn_function(f: NativeFunction) -> bool:
234 def is_py_fft_function(f: NativeFunction) -> bool:
238 def is_py_linalg_function(f: NativeFunction) -> bool:
242 def is_py_nested_function(f: NativeFunction) -> bool:
246 def is_py_sparse_function(f: NativeFunction) -> bool:
250 def is_py_special_function(f: NativeFunction) -> bool:
387 pred: Callable[[NativeFunction], bool],
[all …]
H A Dgen_trace_type.py75 def should_trace(f: NativeFunction) -> bool:
112 def format_trace_op_name(f: NativeFunction) -> str:
141 def format_trace_inputs(f: NativeFunction) -> str:
299 def format_prerecord_trace(f: NativeFunction) -> str:
345 def format_postrecord_trace(f: NativeFunction) -> str:
379 def tie_return_values(f: NativeFunction) -> str:
386 def get_return_value(f: NativeFunction) -> str:
403 def emit_trace_body(f: NativeFunction) -> list[str]:
449 def type_wrapper_name(f: NativeFunction, key: str = "Default") -> str:
466 def method_definition(f: NativeFunction) -> str:
[all …]
H A Dgen_inplace_or_view_type.py276 def inverse_view_name(f: NativeFunction) -> str:
284 def extract_bindings(f: NativeFunction) -> list[Binding]:
300 def unpack_args(f: NativeFunction) -> tuple[list[str], list[Binding]]:
337 def get_base_name(f: NativeFunction) -> str:
341 def get_view_info(f: NativeFunction) -> str | None:
350 f: NativeFunction, bindings: list[Binding], view_idx: str | None = None
521 def modifies_arguments(f: NativeFunction) -> bool:
572 def gen_formals(f: NativeFunction) -> str:
H A Dload_derivatives.py180 def cpp_arguments(f: NativeFunction) -> Sequence[Binding]:
189 f: NativeFunction,
236 f: NativeFunction, formula: str, names: tuple[str, ...]
273 f: NativeFunction,
445 functions_by_signature: dict[FunctionSchema, list[NativeFunction]],
453 functions: Sequence[NativeFunction], name: str
523 f: NativeFunction,
H A Dgen_variable_type.py930 def gen_wrapper_registration(f: NativeFunction, key: str = "Default") -> str:
1111 def gen_differentiable_inputs(f: NativeFunction) -> list[DifferentiableInput]:
1532 f: NativeFunction, input_base: str, unpacked_args: Sequence[str]
1553 f: NativeFunction, unpacked_bindings: list[Binding], var: str
1679 f: NativeFunction, unpacked_bindings: list[Binding], try_jit_decomposition: bool
H A Dgen_view_funcs.py117 f: NativeFunction, include_namespace: bool = False, camel_case: bool = True
146 def returns_multi_tensor(fn: NativeFunction) -> bool:
214 def process_function(fn: NativeFunction, template: CodeTemplate) -> str:
H A Dgen_variable_factories.py63 def is_factory_function(f: NativeFunction) -> bool:
73 def process_function(f: NativeFunction) -> str | None:
/aosp_15_r20/external/pytorch/torchgen/dest/
H A Dlazy_ir.py176 def __call__(self, f: NativeFunctionsGroup | NativeFunction) -> list[str]:
415 def lazy_tensor_decls(self, func: NativeFunction, schema: LazyIrSchema) -> str:
462 func: NativeFunction,
473 def metrics(self, func: NativeFunction, schema: LazyIrSchema) -> str:
476 def get_device(self, func: NativeFunction, schema: LazyIrSchema) -> str:
494 def shape_inference(self, func: NativeFunction, schema: LazyIrSchema) -> str:
569 def build_ir_node(self, func: NativeFunction, schema: LazyIrSchema) -> str:
589 def return_aten_tensor(self, func: NativeFunction, schema: LazyIrSchema) -> str:
620 def __call__(self, func: NativeFunction) -> list[str]:
644 def __init__(self, kernel_name: str, f: NativeFunction, *, symint: bool) -> None:
[all …]
H A Dregister_dispatch_key.py285 def __call__(self, f: NativeFunctionsGroup | NativeFunction) -> list[str]:
303 self, f: NativeFunction
313 self, f: NativeFunction, g: NativeFunctionsGroup | None
393 self, f: NativeFunction, g: NativeFunctionsGroup | None = None
690 f: NativeFunction,
746 def gen_one(self, f: NativeFunction) -> str | None:
H A Dnative_functions.py12 def gen_unstructured(f: NativeFunction, backend_index: BackendIndex) -> str | None:
45 g: NativeFunctionsGroup | NativeFunction, backend_index: BackendIndex
/aosp_15_r20/external/pytorch/torchgen/api/
H A Dpython.py633 def _cpp_signature(f: NativeFunction, *, method: bool = False) -> CppSignature:
637 def has_tensor_options(f: NativeFunction) -> bool:
743 f: NativeFunction, *, method: bool = False, pyi: bool = False
1106 ps: PythonSignature, f: NativeFunction, symint: bool = True
1181 def dispatch_lambda_return_str(f: NativeFunction) -> str:
1210 def cpp_dispatch_target(f: NativeFunction) -> str:
1225 f: NativeFunction,
1378 ps: PythonSignature, f: NativeFunction, *, symint: bool = True
1399 ps: PythonSignature, f: NativeFunction, *, symint: bool = True
H A Dautograd.py312 def is_foreach_func(f: NativeFunction) -> bool:
341 f: NativeFunction,
367 foreach_function: NativeFunction,
583 native_functions: list[NativeFunction],
603 f: NativeFunction,
H A Dunboxing.py101 def name(f: NativeFunction) -> str:
106 def convert_arguments(f: NativeFunction) -> tuple[list[Binding], list[str]]:
/aosp_15_r20/external/pytorch/tools/jit/
H A Dgen_unboxing.py35 def __call__(self, f: NativeFunction) -> str:
99 def __call__(self, f: NativeFunction) -> str:
161 native_functions: Sequence[NativeFunction],
/aosp_15_r20/external/pytorch/torchgen/selective_build/
H A Dselector.py181 def is_native_function_selected(self, func: NativeFunction) -> bool:
211 def is_native_function_selected_for_training(self, func: NativeFunction) -> bool:
349 def op_name_from_native_function(f: NativeFunction) -> str:
/aosp_15_r20/external/pytorch/torchgen/executorch/api/
H A Dcustom_ops.py27 def __call__(self, f: NativeFunction) -> str | None:
85 native_functions: Sequence[NativeFunction],
/aosp_15_r20/external/executorch/backends/vulkan/test/op_tests/
H A Dgenerate_op_correctness_tests.py27 def registry_name(f: NativeFunction) -> str:
43 f_map: Dict[str, NativeFunction], argument
H A Dgenerate_op_benchmarks.py27 def registry_name(f: NativeFunction) -> str:
43 f_map: Dict[str, NativeFunction], argument

12