Go to the documentation of this file.
11 #ifndef PHYSICSTOOLS_ONNXRUNTIME_INTERFACE_ONNXRUNTIME_H_
12 #define PHYSICSTOOLS_ONNXRUNTIME_INTERFACE_ONNXRUNTIME_H_
19 #include "onnxruntime/core/session/onnxruntime_cxx_api.h"
42 int64_t batch_size = 1)
const;
52 static const ::Ort::Env
env_;
ONNXRuntime & operator=(const ONNXRuntime &)=delete
std::unique_ptr<::Ort::Session > session_
static const ::Ort::Env env_
ONNXRuntime(const std::string &model_path, const ::Ort::SessionOptions *session_options=nullptr)
FloatArrays run(const std::vector< std::string > &input_names, FloatArrays &input_values, const std::vector< std::string > &output_names={}, int64_t batch_size=1) const
const std::vector< std::string > & getOutputNames() const
std::vector< std::string > input_node_strings_
std::vector< const char * > input_node_names_
std::vector< std::string > output_node_strings_
std::map< std::string, std::vector< int64_t > > input_node_dims_
const std::vector< int64_t > & getOutputShape(const std::string &output_name) const
std::vector< std::vector< float > > FloatArrays
std::vector< const char * > output_node_names_
std::map< std::string, std::vector< int64_t > > output_node_dims_