Searched refs:native_functions_path (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | gen_autograd.py | 52 native_functions_path: str,
61 os.path.join(autograd_dir, "derivatives.yaml"), native_functions_path, tags_path
66 native_funcs = parse_native_yaml(native_functions_path, tags_path).native_functions
81 native_functions_path,
89 out, native_functions_path, tags_path, fns_with_diff_infos, template_path
98 gen_variable_factories(out, native_functions_path, tags_path, template_path)
105 native_functions_path: str,
111 os.path.join(autograd_dir, "derivatives.yaml"), native_functions_path, tags_path
122 out, native_functions_path, tags_path, deprecated_path, template_path
|
| /aosp_15_r20/external/pytorch/tools/setup_helpers/ |
| H A D | generate_code.py | 24 native_functions_path: str | None = None,
50 native_functions_path or NATIVE_FUNCTIONS_PATH,
61 native_functions_path or NATIVE_FUNCTIONS_PATH,
71 native_functions_path or NATIVE_FUNCTIONS_PATH,
192 options.native_functions_path,
205 aten_path = os.path.dirname(os.path.dirname(options.native_functions_path))
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | codegen_external.py | 20 def gen_external(native_functions_path, tags_path, external_path): argument
21 native_functions = parse_native_yaml(native_functions_path, tags_path)
|
| /aosp_15_r20/external/pytorch/tools/pyi/ |
| H A D | gen_pyi.py | 1485 args.native_functions_path, args.tags_path, args.deprecated_functions_path, fm
|