MCPcopy Index your code
hub / github.com/pytorch/pytorch / set_workspace

Method set_workspace

caffe2/python/net_printer.py:58–71  ·  view source on GitHub ↗
(self, node=None, ws=None, do_copy=False)

Source from the content-addressed store, hash-verified

56
57 @contextmanager
58 def set_workspace(self, node=None, ws=None, do_copy=False):
59 if ws is not None:
60 ws = ws
61 elif node is not None:
62 ws = self.workspaces[str(node)]
63 else:
64 ws = self.workspace
65 if do_copy:
66 ws = copy(ws)
67 self.workspace_ctx.append(ws)
68 try:
69 yield ws
70 finally:
71 del self.workspace_ctx[-1]
72
73 def define_blob(self, blob):
74 self.workspace[blob] += 1

Callers 3

analyze_stepFunction · 0.80
analyze_taskFunction · 0.80
analyze_task_groupFunction · 0.80

Calls 2

copyFunction · 0.50
appendMethod · 0.45

Tested by

no test coverage detected