FM / onnx

/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/onnx/

/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/onnx

Name	Size	Mode	Actions
backend.h	8674	0644	edit dl rm
backend_rep.h	1137	0644	edit dl rm
device.h	488	0644	edit dl rm
helper.h	2819	0644	edit dl rm
offline_tensor.h	1645	0644	edit dl rm
onnxifi_graph_info.h	3539	0644	edit dl rm
onnxifi_init.h	166	0644	edit dl rm
onnx_exporter.h	4843	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/caffe2/onnx/backend_rep.h (1137B)

#pragma once

#include "caffe2/predictor/predictor.h"
#include "caffe2/proto/caffe2_pb.h"

#include <memory>
#include <string>
#include <vector>

namespace caffe2 {
namespace onnx {
class TORCH_API Caffe2BackendRep {
 public:
  void Run(
      const caffe2::Predictor::TensorList& inputs,
      caffe2::Predictor::TensorList* outputs);
  void RunMap(
      const caffe2::Predictor::TensorMap& inputs,
      caffe2::Predictor::TensorList* outputs);

caffe2::NetDef& init_net() {
    return init_net_;
  }
  caffe2::NetDef& pred_net() {
    return pred_net_;
  }
  std::vector<std::string>& uninitialized_inputs() {
    return uninitialized_inputs_;
  }

const caffe2::NetDef& init_net() const {
    return init_net_;
  }
  const caffe2::NetDef& pred_net() const {
    return pred_net_;
  }
  const std::vector<std::string>& uninitialized_inputs() const {
    return uninitialized_inputs_;
  }

private:
  void CheckInit();

caffe2::NetDef init_net_;
  caffe2::NetDef pred_net_;
  std::vector<std::string> uninitialized_inputs_;
  std::unique_ptr<caffe2::Predictor> predictor_{nullptr};
};
} // namespace onnx
} // namespace caffe2

#pragma once #include "caffe2/predictor/predictor.h" #include "caffe2/proto/caffe2_pb.h" #include #include #include namespace caffe2 { namespace onnx { class TORCH_API Caffe2BackendRep { public: void Run( const caffe2::Predictor::TensorList& inputs, caffe2::Predictor::TensorList* outputs); void RunMap( const caffe2::Predictor::TensorMap& inputs, caffe2::Predictor::TensorList* outputs); caffe2::NetDef& init_net() { return init_net_; } caffe2::NetDef& pred_net() { return pred_net_; } std::vector& uninitialized_inputs() { return uninitialized_inputs_; } const caffe2::NetDef& init_net() const { return init_net_; } const caffe2::NetDef& pred_net() const { return pred_net_; } const std::vector& uninitialized_inputs() const { return uninitialized_inputs_; } private: void CheckInit(); caffe2::NetDef init_net_; caffe2::NetDef pred_net_; std::vector uninitialized_inputs_; std::unique_ptr predictor_{nullptr}; }; } // namespace onnx } // namespace caffe2