Searched full:full_codegen (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/torchgen/ |
| D | gen_lazy_tensor.py | 49 # - 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 …]
|
| D | gen_backend_stubs.py | 63 "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
|
| /external/pytorch/aten/src/ATen/native/ |
| D | ts_native_functions.yaml | 3 full_codegen:
|
| /external/pytorch/tools/test/ |
| D | test_gen_backend_stubs.py | 253 …ckend, class_name, cpp_namespace, extra_headers, supported, autograd, full_codegen, non_native, ir…
|
| /external/pytorch/test/lazy/ |
| D | test_ts_opinfo.py | 48 yaml_ts["full_codegen"], yaml_ts["supported"], yaml_ts["autograd"]
|
| /external/pytorch/test/ |
| D | test_meta.py | 1749 ops = d.get("full_codegen", []) + d.get("supported", []) + d.get("autograd", [])
|