NgOXdZddlZddlZddlmZmZddlZgdZdada da da e ej jdZdejvreejdandad ZGd d Zd ZGd dZdZGddZGddZddedefdZddedefdZdS)z& Model / Layer Config singleton state N)AnyOptional) is_exportable is_scriptable is_no_jituse_fused_attnset_exportableset_scriptable set_no_jitset_layer_configset_fused_attnFscaled_dot_product_attentionTIMM_FUSED_ATTNctSN)_NO_JITN/var/www/html/ai-engine/env/lib/python3.11/site-packages/timm/layers/config.pyrr%s Nrc6eZdZdeddfdZddZdedefdZdS) r modereturnNc"t|_|adSr)rprevselfrs r__init__zset_no_jit.__init__*s rcdSrrrs r __enter__zset_no_jit.__enter__/ rargsc|jadSNF)rrrr#s r__exit__zset_no_jit.__exit__2s)urrN__name__ __module__ __qualname__boolrr!rr'rrrr r )slTd     cdrr ctSr) _EXPORTABLErrrrr8 rc6eZdZdeddfdZddZdedefdZdS) r rrNc"t|_|adSr)r/rrs rrzset_exportable.__init__=  rcdSrrr s rr!zset_exportable.__enter__Br"rr#c|jadSr%)rr/r&s rr'zset_exportable.__exit__Ei urr(r)rrrr r <lTd     cdrr ctSr) _SCRIPTABLErrrrrKr0rc6eZdZdeddfdZddZdedefdZdS) r rrNc"t|_|adSr)r9rrs rrzset_scriptable.__init__Pr3rcdSrrr s rr!zset_scriptable.__enter__Ur"rr#c|jadSr%)rr9r&s rr'zset_scriptable.__exit__Xr6rr(r)rrrr r Or7rr c |eZdZdZ d deedeedeedeefdZd d Zd edefd Z dS)r z Layer config context manager that allows setting all layer config flags at once. If a flag arg is None, it will not change the current value. N scriptable exportableno_jitno_activation_jitchttttf|_||a||a||a||adSdSr)r9r/r_NO_ACTIVATION_JITr)rr?r@rArBs rrzset_layer_config.__init__bsP g7II  !$K  !$K  G  (!2    ) (rrcdSrrr s rr!zset_layer_config.__enter__vr"rr#c|j\aaaadSr%)rr9r/rrDr&s rr'zset_layer_config.__exit__ys AE = ['+=ur)NNNNr() r*r+r,__doc__rr-rr!rr'rrrr r ^s *.)-%)04 33 3!3TN 3 (~ 3333(    cdrr experimentalrcRtrtrdS|r tdkStdkS)NFrr)_HAS_FUSED_ATTNr/_USE_FUSED_ATTN)rHs rrrs4 ku#"" Q rTenablec`tstjddS|r|rdadS|rdadSdadS)Nz^This version of pytorch does not have F.scaled_dot_product_attention, fused_attn flag ignored.rr)rJwarningswarnrK)rLrHs rr r sR  vwww r)F)TF)rGosrOtypingrrtorch__all__rrDr/r9hasattrnn functionalrJenvironintrKrr rr rr r r-rr rrrrZs         '%(-/MNN ""c"*%6788OOO                        !!!!!!!!H$  4 d      r