Searched full:ir_gen (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | gen_lazy_tensor.py | 104 ir_gen = yaml_values.pop("ir_gen", [])
107 assert isinstance(ir_gen, list)
109 ir_gen_opnames = [OperatorName.parse(name) for name in ir_gen]
362 # for ir_gen, we generate only IR
365 full_codegen, non_native, ir_gen = parse_native_functions_keys(
547 lazy_ir_obj, grouped_native_functions, full_codegen + ir_gen
|
| H A D | gen_backend_stubs.py | 65 "ir_gen",
115 # ir_gen is ignored by parse_backend_yaml, and re-parsed in gen_lazy_tensor.py
116 yaml_values.pop("ir_gen", {})
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ts_native_functions.yaml | 174 ir_gen:
|
| /aosp_15_r20/external/pytorch/tools/test/ |
| H A D | test_gen_backend_stubs.py | 253 …namespace, extra_headers, supported, autograd, full_codegen, non_native, ir_gen, symint""", # noq…
|