U ‰d'ã@s„ddlZddlZddlmZddlmZdZe ¡add„Z dd„Z ej dd d „ƒZ ej dd d „ƒZ ej dd„ƒZej dd„ƒZdS)éN)Ú basestring)Ú caffe2_pb2ú/cCsttdƒsdt_tjS)NÚ namescopeÚ)ÚhasattrÚ_threadlocal_scoper©r r ú7/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/scope.pyÚCurrentNameScopes r cCsttdƒsdt_tjS)NÚ devicescope)rrr r r r r ÚCurrentDeviceScopes r Fc cstt|tƒs|dkstdƒ‚tƒ}|r,|tnd}|r<|t_n tj|t_z dVW5tj |¡shtdƒ‚|t_XdS)Nz,NameScope takes in a string as its argument.rzAThe namescope variable is changed from outside NameScope() calls.)Ú isinstancerÚAssertionErrorr Ú_NAMESCOPE_SEPARATORrrÚendswith)ÚprefixÚresetÚ old_scoper r r Ú NameScope#sÿ  ÿrccs¾t ¡}|r,t|tjƒs tdƒ‚| |¡n |s8tdƒ‚|rB||_tƒ}|rh| d¡rh| d¡sh|j|_|r„t|dƒr„|j   |j ¡|j   ¡|t _ z dVW5t j |ks²tdƒ‚|t _ XdS)Nz?DeviceScope takes in a caffe2_pb2.DeviceOption as its argument.z7At least one argument should be non-null in DeviceScopeÚ node_nameÚ extra_infoz=The device scope is changed from outside DeviceScope() calls.)rZ DeviceOptionrrZCopyFromrr ZHasFieldrrÚextendÚsortrr )ZscoperZ new_scoperr r r Ú DeviceScope7s0ÿ  ÿ   ÿrc cs,tƒ}dzdt_dVW5|t_£dSX)z© Allow users to 'disable' the name scope behaviour. This sets the CurrentNameScope() to None, so that the field is not set in CreateOperator(...), etc. Nr)r rr©rr r r ÚEmptyNameScopeYs  rc cs,tƒ}dzdt_dVW5|t_£dSX)a" Allow users to 'disable' the device scope behaviour (so it can be controlled at a NetDef::DeviceOption level, not overridden at OperatorDef::DeviceOption level). This sets the CurrentDeviceScope() to None, so that the field is not set in CreateOperator(...), etc. N)r rr rr r r ÚEmptyDeviceScopejs  r)F)N)Ú contextlibÚ threadingZ past.builtinsrZ caffe2.protorrÚlocalrr r Úcontextmanagerrrrrr r r r Ús    !