コード例 #1
0
def RunPlan(plan_or_step):
    # TODO(jiayq): refactor core.py/workspace.py to avoid circular deps
    import caffe2.python.core as core
    if isinstance(plan_or_step, core.ExecutionStep):
        plan_or_step = core.Plan(plan_or_step)
    return C.run_plan(StringifyProto(plan_or_step))
コード例 #2
0
ファイル: workspace.py プロジェクト: gtgalone/pytorch
def RunPlan(plan_or_step):
    # TODO(jiayq): refactor core.py/workspace.py to avoid circular deps
    import caffe2.python.core as core
    if isinstance(plan_or_step, core.ExecutionStep):
        plan_or_step = core.Plan(plan_or_step)
    return C.run_plan(StringifyProto(plan_or_step))
コード例 #3
0
ファイル: workspace.py プロジェクト: slayton58/caffe2
def RunPlan(plan):
    return C.run_plan(StringfyProto(plan))