13 #include "onnxruntime_cxx_api.h"
14 #ifndef HEPDATAINCLUDEDIR
15 #define HEPDATAINCLUDEDIR ""
27 Ort::Session* _session;
28 Ort::AllocatorWithDefaultOptions allocator;
29 std::vector<Ort::AllocatedStringPtr> input_names_ptr;
31 std::vector<int64_t> input_node_dims;
33 std::vector<Ort::AllocatedStringPtr> output_names_ptr;
35 std::vector<int64_t> output_node_dims;
42 HEPONNX(std::string filename);
44 void addonnx(std::string filename);
46 float get_result(std::vector<double> &nn_inputs);
50 size_t input_tensor_size;
51 std::vector<const char*> input_node_names;
52 size_t num_input_nodes;
53 size_t output_tensor_size;
54 std::vector<const char*> output_node_names;
55 size_t num_output_nodes;