˲j" VdZddlmZGddZGddejZdS)z3Functionality for Python <-> C++ frontend inter-op.)nncXeZdZdZdZedZdZdZdZ dZ dZ d Z d Z d S) OrderedDictWrapperaA wrapper around a C++ OrderedDict. It dynamically evaluates the OrderedDict getter on a bound C++ module, such that new changes on the C++ side are picked up. Otherwise accessing e.g. ``cpp_module._parameters`` just once would get a frozen copy of the parameters at the time of access. ``torch.nn.Module`` accesses ``_parameters`` et al. via ``self.__dict__`` so using properties does not work. c"||_||_dSN) cpp_moduleattr)selfrr s =C:\Projects\Historical\venv\Lib\site-packages\torch/nn/cpp.py__init__zOrderedDictWrapper.__init__s$ c6t|j|jSr)getattrrr r s r cpp_dictzOrderedDictWrapper.cpp_dictst 222r c4|jSr)ritemsrs r rzOrderedDictWrapper.itemss}""$$$r c4|jSr)rkeysrs r rzOrderedDictWrapper.keyss}!!###r c4|jSr)rvaluesrs r rzOrderedDictWrapper.values"s}##%%%r c4|jSr)r__iter__rs r rzOrderedDictWrapper.__iter__%s}%%'''r c4|jSr)r__len__rs r rzOrderedDictWrapper.__len__(s}$$&&&r c6|j|Sr)r __contains__r keys r rzOrderedDictWrapper.__contains__+s}))#...r c6|j|Sr)r __getitem__rs r r!zOrderedDictWrapper.__getitem__.s}((---r N)__name__ __module__ __qualname____doc__r propertyrrrrrrrr!r r rrs33X3 %%%$$$&&&((('''///.....r rcfeZdZdZfdZddZedZejdZdZ xZ S) ModuleWrapperz\A subclass of ``torch.nn.Module`` that wraps a C++ frontend module and delegates all access.c j||_tt|d|_t|d|_t|d|_t|D];}|ds$t||t|j|z:,V,V ,>z:,V,V  OO D DD??3'' DdGDOT$B$BCCC D Dr Tc|D]=}||j|_|j||jj|j_>|D]}||j|_|Sr) parametersdata_gradbuffers)r fnrecurseparambufs r _applyzModuleWrapper._applyBs__&& 8 8EEJEJ{&#%2ek&6#7#7  <<>> $ $Cr#(||CHH r c|jjSr)rtrainingrs r r?zModuleWrapper.trainingPs ''r c:|j|dSr)rtrain)r modes r r?zModuleWrapper.trainingTs d#####r c4|jSr)r__repr__rs r rDzModuleWrapper.__repr__Xs'')))r )T) r"r#r$r%r r=r&r?setterrD __classcell__)r3s@r r)r)2sff D D D D D    ((X(_$$_$*******r r)N)r%torchrrModuler)r'r r rIs|99(.(.(.(.(.(.(.(.V'*'*'*'*'*BI'*'*'*'*'*r