a ==ic @s8ddlZddlmZGdddeZGdddeZdS)N) workspacec@s$eZdZddZddZddZdS) _WorkspaceCtxcCs||_g|_dSN) workspace_idworkspace_stack)selfrrm/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/caffe2/python/onnx/workspace.py__init__sz_WorkspaceCtx.__init__cCs$|jttj|jdddSNT)Zcreate_if_missing)rappendrZCurrentWorkspaceSwitchWorkspacerrrrr __enter__sz_WorkspaceCtx.__enter__cCs|j}tj|dddSr )rpoprr )rexc_type exc_value tracebackwrrr __exit__s z_WorkspaceCtx.__exit__N)__name__ __module__ __qualname__r rrrrrr rsrc@s(eZdZdZddZddZddZdS) WorkspaceaA An object representing a Caffe2 workspace. It is a context manager, so you can say 'with workspace:' to use the represented workspace as your global workspace. It also supports every method supported by caffe2.python.workspace, but instead of running these operations in the global workspace, it runs them in the workspace represented by this object. When this object goes dead, the workspace (and all nets and blobs within it) are freed. Why do we need this class? Caffe2's workspace model is very "global state" oriented, in that there is always some ambient global workspace you are working in which holds on to all of your networks and blobs. This class makes it possible to work with workspaces more locally, and without forgetting to deallocate everything in the end. cCsttt|_dSr)rstruuiduuid4_ctxrrrr r 5szWorkspace.__init__csfdd}|S)Ncs>j$tt|i|WdS1s00YdSr)rgetattrr)argskwargsattrrrr f=sz Workspace.__getattr__..fr)rr"r#rr!r __getattr__<szWorkspace.__getattr__cCs |dSr)ZResetWorkspacerrrr __del__BszWorkspace.__del__N)rrr__doc__r r$r%rrrr r%sr)rZ caffe2.pythonrobjectrrrrrr  s