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

Method _compile_task_group

caffe2/python/session.py:190–195  ·  view source on GitHub ↗
(cls, task_group, setup_net_list=None)

Source from the content-addressed store, hash-verified

188
189 @classmethod
190 def _compile_task_group(cls, task_group, setup_net_list=None):
191 with Cluster():
192 task = task_group.to_task()
193 plan = core.Plan('task_group_plan')
194 plan.AddStep(task.get_step())
195 return (plan, task.output_list(), task.workspace_type())
196
197 def _run_compiled(self, compiled):
198 plan, output_list, workspace_type = compiled

Callers

nothing calls this directly

Calls 6

AddStepMethod · 0.95
ClusterClass · 0.90
to_taskMethod · 0.80
get_stepMethod · 0.80
output_listMethod · 0.80
workspace_typeMethod · 0.45

Tested by

no test coverage detected