3 #include <unordered_set> 4 #include "caffe2/core/net.h" 5 #include "caffe2/core/tensor.h" 6 #include "caffe2/proto/metanet.pb.h" 7 #include "caffe2/proto/predictor_consts.pb.h" 13 using TensorVector = std::vector<TensorCPU*>;
14 using TensorMap = std::unordered_map<std::string, TensorCPU*>;
23 const NetDef& init_net,
24 const NetDef& run_net,
39 bool run(
const TensorVector& inputs, TensorVector* outputs);
42 bool run_map(
const TensorMap& inputs, TensorVector* outputs);
44 const NetDef& def()
const {
55 std::unordered_set<std::string> inputNames_;
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 ...