o NeÝ ã@sBddlZddlZddlmZmZGdd„dƒZdd„Zdd„ZdS) éN)Údefault_main_programÚ_non_static_modec@seZdZdZddd„ZdS)ÚLayerOpsRecoderz= Record generated operators information in nn.Layer. éÿÿÿÿNFcCs"||_||_||_||_||_dS)N)ÚstartÚendÚopsÚis_validÚhooks)Úselfrrrr r ©r úPD:\Projects\ConvertPro\env\Lib\site-packages\paddle/fluid/dygraph/layer_hooks.pyÚ__init__s  zLayerOpsRecoder.__init__)rrNFN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rr r r r rsrcCsPtƒs&|jjdkrttƒ ¡jƒ|j_d|j_dSd|j_t  d  |j ¡¡dS)zC A pre-hook to mark op numbers before enter layer.forward. rTFzZ{} has recorded the op information before. Please check whether you call this layer twice.N) rÚ _op_recorderrÚlenrÚ current_blockrr ÚwarningsÚwarnÚformatZ _full_name)ÚlayerÚinputsr r r Úrecord_program_ops_pre_hook"s  ÿûþrc Cs¤tƒsP|jjrP|jj}ttƒ ¡jƒ}|dkr||ksJ‚tƒ ¡j||…}||j_||j_|D]}t   |j ¡D] \}}|  ||¡q9q1|jj D]} |  ¡qIdS)zd A post-hook to append customized attributes into all operators generated in current layer. rN)rrr rrrrrrÚsixÚ iteritemsZ_customized_attrsZ _set_attrr Úremove) rrZoutputsrrrÚopZ attr_nameÚvalZ hook_helperr r r Ú!set_op_customized_attrs_post_hook4sÿ  r!)rrZpaddle.fluid.frameworkrrrrr!r r r r Ús