Class OnnxInferenceEngine#

Class Documentation#

class OnnxInferenceEngine#

Public Functions

OnnxInferenceEngine()#
~OnnxInferenceEngine()#
OnnxInferenceEngine(const OnnxInferenceEngine&) = delete#
OnnxInferenceEngine &operator=(const OnnxInferenceEngine&) = delete#
OnnxInferenceEngine(OnnxInferenceEngine&&) noexcept#
OnnxInferenceEngine &operator=(OnnxInferenceEngine&&) noexcept#
std::expected<void, OnnxError> initialize(const SessionConfig &config)#
bool is_initialized() const#
std::expected<InferenceResult, OnnxError> run_inference(std::span<const float> input_data, const TensorShape &input_shape, const std::string &input_name = "input")#
std::expected<InferenceResult, OnnxError> run_inference_multi_input(const std::vector<std::pair<std::string, TensorData>> &inputs)#
std::vector<std::string> get_input_names() const#
std::vector<std::string> get_output_names() const#
std::expected<TensorShape, OnnxError> get_input_shape(const std::string &name) const#
std::expected<TensorShape, OnnxError> get_output_shape(const std::string &name) const#