Home
last modified time | relevance | path

Searched refs:tensorimpl_key (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/optim/
H A Dserialize.h24 std::string tensorimpl_key = in serialize() local
29 archive.write(tensorimpl_key, param_state_archive); in serialize()
39 for (const std::string& tensorimpl_key : tensorimpl_keys) { in serialize() local
41 archive.read(tensorimpl_key, param_state_archive); in serialize()
44 state[reinterpret_cast<void*>(std::stoull(tensorimpl_key))] = in serialize()