1 #ifndef CAFFE2_CORE_NET_SIMPLE_ASYNC_H_ 2 #define CAFFE2_CORE_NET_SIMPLE_ASYNC_H_ 6 #include "caffe2/core/common.h" 7 #include "caffe2/core/logging.h" 8 #include "caffe2/core/net.h" 9 #include "caffe2/core/registry.h" 10 #include "caffe2/core/tensor.h" 11 #include "caffe2/core/workspace.h" 12 #include "caffe2/proto/caffe2.pb.h" 23 bool SupportsAsync()
override {
28 const int warmup_runs,
30 const bool run_individual)
override;
38 vector<OperatorBase*> GetOperators()
const override {
39 vector<OperatorBase*> op_list;
40 for (
auto& op : operators_) {
41 op_list.push_back(op.get());
47 bool DoRunAsync()
override;
49 vector<unique_ptr<OperatorBase>> operators_;
56 #endif // CAFFE2_CORE_NET_SIMPLE_ASYNC_H_ vector< float > TEST_Benchmark(const int warmup_runs, const int main_runs, const bool run_individual) override
Benchmarks a network.
Workspace is a class that holds all the related objects created during runtime: (1) all blobs...
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...