Function print_task_group

caffe2/python/net_printer.py:384–387

(text, tg, header=None)

@Printer.register(TaskGroup)
def print_task_group(text, tg, header=None):
    with text.context(header or call('TaskGroup')):
        for task in tg.tasks_by_node().tasks():
            text(task)


@Printer.register(Job)