1 #ifndef CAFFE2_CORE_NET_SIMPLE_H_ 2 #define CAFFE2_CORE_NET_SIMPLE_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" 22 bool SupportsAsync()
override {
27 const int warmup_runs,
29 const bool run_individual)
override;
37 vector<OperatorBase*> GetOperators()
const override {
38 vector<OperatorBase*> op_list;
39 for (
auto& op : operators_) {
40 op_list.push_back(op.get());
47 bool RunAsync()
override;
49 vector<unique_ptr<OperatorBase>> operators_;
56 #endif // CAFFE2_CORE_NET_SIMPLE_H_
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 ...
vector< float > TEST_Benchmark(const int warmup_runs, const int main_runs, const bool run_individual) override
Benchmarks a network.