B #=6\¸ã@s´ddlmZddlmZddlmZddlmZddlZddlZddlmZddl m Z dZ e  ¡a d d „Zd d „Zejddd„ƒZejddd„ƒZejdd„ƒZejdd„ƒZdS)é)Úabsolute_import)Údivision)Úprint_function)Úunicode_literalsN)Ú basestring)Ú caffe2_pb2ú/cCsttdƒsdt_tjS)NÚ namescopeÚ)ÚhasattrÚ_threadlocal_scoper ©r r ú6/tmp/pip-install-l3r2oljg/torch/caffe2/python/scope.pyÚCurrentNameScopes rcCsttdƒsdt_tjS)NÚ devicescope)r r rr r r rÚCurrentDeviceScopes rFc cstt|tƒs|dkstdƒ‚tƒ}|r,|tnd}|r<|t_n tj|t_z dVWdtj |¡shtdƒ‚|t_XdS)Nz,NameScope takes in a string as its argument.r zAThe namescope variable is changed from outside NameScope() calls.)Ú isinstancerÚAssertionErrorrÚ_NAMESCOPE_SEPARATORr r Ú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 dVWdt 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 DeviceOptionrrZCopyFromrrZHasFieldr rÚextendÚsortr r)ÚscoperZ new_scoperr r rÚ DeviceScope7s*      rccs"tƒ}zdt_dVWd|t_dS)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. r N)rr r )rr r rÚEmptyNameScopeYs  r ccs"tƒ}zdt_dVWd|t_dS)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)rr r)rr r rÚEmptyDeviceScopejs  r!)F)N)Ú __future__rrrrÚ contextlibÚ threadingZ past.builtinsrZ caffe2.protorrÚlocalr rrÚcontextmanagerrrr r!r r r rÚs"        !