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

Method add

caffe2/python/task.py:208–221  ·  view source on GitHub ↗
(self, task)

Source from the content-addressed store, hash-verified

206 return self._remote_nets
207
208 def add(self, task):
209 assert not self._already_used, (
210 'Cannot add Task to an already used TaskGroup.')
211 assert (
212 self._workspace_type is None or
213 task._workspace_type is None or
214 self._workspace_type == task._workspace_type)
215 if task._workspace_type is None:
216 task._workspace_type = (
217 self._workspace_type or WorkspaceType.PRIVATE)
218 if self._workspace_type is None:
219 self._workspace_type = task._workspace_type
220 task._notify_used()
221 self._tasks.append(task)
222
223 def tasks(self):
224 for task in self._tasks_to_add:

Callers 15

tasksMethod · 0.95
compileMethod · 0.95
_make_unique_nameFunction · 0.45
_operators_to_graph_defFunction · 0.45
test_small_sls_acc32Method · 0.45
test_dequantizeMethod · 0.45
test_quantizeMethod · 0.45
test_layernormMethod · 0.45
test_batch_matmulMethod · 0.45
test_bnMethod · 0.45

Calls 2

_notify_usedMethod · 0.80
appendMethod · 0.45

Tested by 15

test_small_sls_acc32Method · 0.36
test_dequantizeMethod · 0.36
test_quantizeMethod · 0.36
test_layernormMethod · 0.36
test_batch_matmulMethod · 0.36
test_bnMethod · 0.36
test_small_slsMethod · 0.36