3i RdZddlmZGddZGddejZy)z3Functionality for Python <-> C++ frontend inter-op.)nnc^eZdZdZd dZedZdZdZdZ d Z de fd Z de fd Zd Zy)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. returnNc ||_||_yN) cpp_moduleattr)selfr r s N/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/torch/nn/cpp.py__init__zOrderedDictWrapper.__init__s$ cBt|j|jSr)getattrr r r s r cpp_dictzOrderedDictWrapper.cpp_dictst 22rc6|jjSr)ritemsrs r rzOrderedDictWrapper.itemss}}""$$rc6|jjSr)rkeysrs r rzOrderedDictWrapper.keyss}}!!##rc6|jjSr)rvaluesrs r rzOrderedDictWrapper.values"s}}##%%rc6|jjSr)r__iter__rs r rzOrderedDictWrapper.__iter__%s}}%%''rc6|jjSr)r__len__rs r rzOrderedDictWrapper.__len__(s}}$$&&rc8|jj|Sr)r __contains__r keys r rzOrderedDictWrapper.__contains__+s}}))#..rc8|jj|Sr)r __getitem__rs r r"zOrderedDictWrapper.__getitem__.s}}((--rrN)__name__ __module__ __qualname____doc__r propertyrrrrrintrboolrr"rr rrsQ33 %$&(''/4/.rrcneZdZdZdfd Zd dZedZejddZde fdZ xZ S) ModuleWrapperz\A subclass of ``torch.nn.Module`` that wraps a C++ frontend module and delegates all access.rc ||_t| t|d|_t|d|_t|d|_t|D]5}|jdrt||t|j|7y)N _parameters_buffers_modules_) r superr rr/r0r1dir startswithsetattrr)r r r __class__s r r zModuleWrapper.__init__5sz% -j-H,>z:,V ,>z:,V  O DD??3'dGDOOT$BC Drc&|jD]Q}||j|_|j'||jj|j_S|jD]}||j|_|Sr) parametersdata_gradbuffers)r fnrecurseparambufs r _applyzModuleWrapper._applyBs{__& 8EEJJEJ{{&#%ekk&6&6#7   8<<> $C#((|CH $ rc.|jjSr)r trainingrs r rCzModuleWrapper.trainingPs'''rc:|jj|yr)r train)r modes r rCzModuleWrapper.trainingUs d#rc6|jjSr)r __repr__rs r rHzModuleWrapper.__repr__Ys''))rr#)T) r$r%r&r'r rAr(rCsetterstrrH __classcell__)r7s@r r-r-2sIf D ((__$$*#*rr-N)r'torchrrModuler-r+rr rNs&9(.(.V(*BII(*r