3 #include "caffe2/predictor/predictor.h" 4 #include "caffe2/proto/caffe2_pb.h" 14 const caffe2::Predictor::TensorList& inputs,
15 caffe2::Predictor::TensorList* outputs);
17 const caffe2::Predictor::TensorMap& inputs,
18 caffe2::Predictor::TensorList* outputs);
20 caffe2::NetDef& init_net() {
23 caffe2::NetDef& pred_net() {
26 std::vector<std::string>& uninitialized_inputs() {
27 return uninitialized_inputs_;
30 const caffe2::NetDef& init_net()
const {
33 const caffe2::NetDef& pred_net()
const {
36 const std::vector<std::string>& uninitialized_inputs()
const {
37 return uninitialized_inputs_;
43 caffe2::NetDef init_net_;
44 caffe2::NetDef pred_net_;
45 std::vector<std::string> uninitialized_inputs_;
46 std::unique_ptr<caffe2::Predictor> predictor_{
nullptr};
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...