mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
7f38ea4555
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/9293 Reviewed By: huitseeker Differential Revision: D8778499 fbshipit-source-id: 0cf59e02cb37b3fe22885c1b5e10b5d2e7585382
14 lines
219 B
C++
14 lines
219 B
C++
#pragma once
|
|
|
|
#include <functional>
|
|
|
|
namespace caffe2 {
|
|
|
|
class Workspace;
|
|
class PlanDef;
|
|
|
|
typedef std::function<bool(int)> ShouldContinue;
|
|
|
|
bool RunPlanOnWorkspace(Workspace* ws, const PlanDef& plan, ShouldContinue);
|
|
}
|