Searched defs:map_location (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _serialization.py | 90 def load(f, map_location=None, _extra_files=None, _restore_shapes=False): argument
175 def validate_map_location(map_location=None): argument
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | serialization.py | 1155 def _legacy_load(f, map_location, pickle_module, **pickle_load_args): argument
1399 def _get_restore_location(map_location): argument
1433 def _load(zip_file, map_location, pickle_module, pickle_file='data.pkl', overall_storage=None, **pi… argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | script_init.cpp | 1844 bool restore_shapes = false) { in initJitScriptBindings()
1887 py::object map_location, in initJitScriptBindings()
1910 [](const std::string& filename, py::object map_location) { in initJitScriptBindings()
1921 [](const std::string& buffer, py::object map_location) { in initJitScriptBindings()
|
| /aosp_15_r20/external/pytorch/torch/jit/mobile/ |
| H A D | __init__.py | 8 def _load_for_lite_interpreter(f, map_location=None): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_jit.py | 165 def getExportImportCopy(self, m, also_test_file=True, map_location=None): argument
|
| H A D | jit_utils.py | 269 def getExportImportCopyWithPacking(self, m, also_test_file=True, map_location=None): argument
|
| /aosp_15_r20/external/pytorch/torch/package/ |
| H A D | package_importer.py | 223 def load_pickle(self, package: str, resource: str, map_location=None) -> Any: argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_serialization.py | 495 def map_location(storage, loc): function
|