Caffe2 - C++ API
A deep learning, cross platform ML framework
plan_executor.h
1 
17 #pragma once
18 
19 #include <functional>
20 #ifndef CAFFE2_MOBILE
21 #include "caffe2/core/stats.h"
22 #endif // CAFFE2_MOBILE
23 
24 namespace caffe2 {
25 
26 class Workspace;
27 class PlanDef;
28 
29 typedef std::function<bool(int)> ShouldContinue;
30 
31 bool RunPlanOnWorkspace(Workspace* ws, const PlanDef& plan, ShouldContinue);
32 
33 #ifndef CAFFE2_MOBILE
35  CAFFE_STAT_CTOR(PlanExecutionTime);
36  CAFFE_EXPORTED_STAT(plan_execution_time_ns);
37 };
38 #endif // CAFFE2_MOBILE
39 }
Copyright (c) 2016-present, Facebook, Inc.