mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
7f38ea4555
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/9293 Reviewed By: huitseeker Differential Revision: D8778499 fbshipit-source-id: 0cf59e02cb37b3fe22885c1b5e10b5d2e7585382
13 lines
219 B
C++
13 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);
|
|
}
|