pytorch/torch/csrc/jit/api
Zino Benaissa 11cc7f143d Run __setstate__ when cloning modules (#45858)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/45858

When cloning a module that has __setstate__, __getstate__ methods.
We need to load these methods to initialize these modules.

Test Plan: Imported from OSS

Reviewed By: suo

Differential Revision: D24116524

Pulled By: bzinodev

fbshipit-source-id: a5111638e2dc903781f6468838c000850d1f9a74
2020-10-16 15:55:31 -07:00
..
compilation_unit.h [JIT] Add property support to TorchScript classes (#42389) 2020-08-14 12:56:57 -07:00
function_impl.cpp [JIT] Dont optimize shape peepholes on inline (#36404) 2020-04-15 17:49:48 -07:00
function_impl.h [model transform] tuple to arglist jit pass (#36093) 2020-04-09 22:05:43 -07:00
method.h [JIT] clang-format JIT code (#35115) 2020-03-26 11:24:51 -07:00
module_save.cpp Get, save, and load module information for each operator (#42133) 2020-08-14 01:25:27 -07:00
module.cpp Run __setstate__ when cloning modules (#45858) 2020-10-16 15:55:31 -07:00
module.h Get, save, and load module information for each operator (#42133) 2020-08-14 01:25:27 -07:00
object.cpp [jit][api][refactor] remove redundant deepcopy implementation (#37538) 2020-05-06 15:41:33 -07:00
object.h [JIT] Make objects throw Python AttributeError on nonexistant attr access (#45911) 2020-10-07 01:57:29 -07:00