Home
last modified time | relevance | path

Searched full:full_codegen (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torchgen/
H A Dgen_lazy_tensor.py49 # - opt-in: 'full_codegen' section of backend yaml; 'supported' section omits these implementations
50 …iveFunctions.h declares implementations of native functions for both 'supported' and 'full_codegen'
71 # - `full_codegen` ops get implementations (and IR classes) generated too
87 …s a Tuple of (backend_key, autograd_key, cpp_namespace, updated BackendIndex mapping, full_codegen)
90 ["backend_key", "autograd_key", "cpp_namespace", "backend_indices", "full_codegen"],
102 full_codegen = yaml_values.pop("full_codegen", [])
105 assert isinstance(full_codegen, list)
108 full_codegen_opnames = [OperatorName.parse(name) for name in full_codegen]
361 # for full_codegen, we generate IR, kernels, etc
365 full_codegen, non_native, ir_gen = parse_native_functions_keys(
[all …]
H A Dgen_backend_stubs.py63 "full_codegen",
108 # full_codegen is ignored by parse_backend_yaml, and re-parsed in gen_lazy_tensor.py
109 full_codegen = yaml_values.pop("full_codegen", [])
110 supported.extend(full_codegen)
255 full_codegen: list[OperatorName] | None = None,
265 if full_codegen is None:
266 full_codegen = []
287 and f.func.name not in full_codegen
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dts_native_functions.yaml3 full_codegen:
/aosp_15_r20/external/pytorch/tools/test/
H A Dtest_gen_backend_stubs.py253 …ckend, class_name, cpp_namespace, extra_headers, supported, autograd, full_codegen, non_native, ir…
/aosp_15_r20/external/pytorch/test/lazy/
H A Dtest_ts_opinfo.py48 yaml_ts["full_codegen"], yaml_ts["supported"], yaml_ts["autograd"]
/aosp_15_r20/external/pytorch/test/
H A Dtest_meta.py1749 ops = d.get("full_codegen", []) + d.get("supported", []) + d.get("autograd", [])