1 #include "caffe2/core/common.h" 2 #include "caffe2/onnx/backend_rep.h" 6 namespace caffe2 {
namespace onnx {
8 void Caffe2BackendRep::CheckInit() {
10 predictor_ = caffe2::make_unique<caffe2::Predictor>(init_net_, pred_net_);
17 void Caffe2BackendRep::Run(
18 const caffe2::Predictor::TensorVector& inputs,
19 caffe2::Predictor::TensorVector* outputs) {
21 predictor_->run(inputs, outputs);
24 void Caffe2BackendRep::RunMap(
25 const caffe2::Predictor::TensorMap& inputs,
26 caffe2::Predictor::TensorVector* outputs) {
28 predictor_->run_map(inputs, outputs);
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...