tj VdZddlmZGddZGddejZdS)z3Functionality for Python <-> C++ frontend inter-op.)nncfeZdZdZd dZedZdZdZdZ d Z de fd Z de fd Zd ZdS)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"||_||_dSN) cpp_moduleattr)selfr r s J/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/nn/cpp.py__init__zOrderedDictWrapper.__init__s$ c6t|j|jSr)getattrr r r s r cpp_dictzOrderedDictWrapper.cpp_dictst 222rc4|jSr)ritemsrs r rzOrderedDictWrapper.itemss}""$$$rc4|jSr)rkeysrs r rzOrderedDictWrapper.keyss}!!###rc4|jSr)rvaluesrs r rzOrderedDictWrapper.values"s}##%%%rc4|jSr)r__iter__rs r rzOrderedDictWrapper.__iter__%s}%%'''rc4|jSr)r__len__rs r rzOrderedDictWrapper.__len__(s}$$&&&rc6|j|Sr)r __contains__r keys r rzOrderedDictWrapper.__contains__+s}))#...rc6|j|Sr)r __getitem__rs r r"zOrderedDictWrapper.__getitem__.s}((---rrN)__name__ __module__ __qualname____doc__r propertyrrrrrintrboolrr"rr rrs33X3 %%%$$$&&&((('''''/4////.....rrcpeZdZdZd fd Zd dZedZejd dZde fd Z xZ S) ModuleWrapperz\A subclass of ``torch.nn.Module`` that wraps a C++ frontend module and delegates all access.rNc 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 DrTc|D]=}||j|_|j||jj|j_>|D]}||j|_|Sr) parametersdata_gradbuffers)r fnrecurseparambufs r _applyzModuleWrapper._applyBs__&& 8 8EEJEJ{&#%2ek&6#7#7  <<>> $ $Cr#(||CHH rc|jjSr)r trainingrs r rCzModuleWrapper.trainingPs''rc:|j|dSr)r train)r modes r rCzModuleWrapper.trainingUs d#####rc4|jSr)r __repr__rs r rHzModuleWrapper.__repr__Ys'')))rr#)T) r$r%r&r'r rAr(rCsetterstrrH __classcell__)r7s@r r-r-2sff D D D D D D    ((X(_$$$_$*#********rr-N)r'torchrrModuler-r+rr rNs|99(.(.(.(.(.(.(.(.V(*(*(*(*(*BI(*(*(*(*(*r