File OnnxRuntimeBase.hpp¶
-
namespace Acts
Note
This file is foreseen for the
Geometry
module to replaceExtent
Note
this is foreseen for the ‘Geometry’ module
-
class OnnxRuntimeBase
- #include </home/docs/checkouts/readthedocs.org/user_builds/acts/checkouts/v23.0.0/Plugins/Onnx/include/Acts/Plugins/Onnx/OnnxRuntimeBase.hpp>
Subclassed by Acts::MLTrackClassifier
Public Functions
-
OnnxRuntimeBase() = default
Default constructor.
-
OnnxRuntimeBase(Ort::Env &env, const char *modelPath)
Parametrized constructor.
- Parameters
env – the ONNX runtime environment
modelPath – the path to the ML model in *.onnx format
-
~OnnxRuntimeBase() = default
Default destructor.
Protected Functions
-
std::vector<float> runONNXInference(std::vector<float> &inputTensorValues) const¶
Run the ONNX inference function.
- Parameters
inputTensorValues – The input feature values used for prediction
- Returns
The output (predicted) values
-
OnnxRuntimeBase() = default
-
class OnnxRuntimeBase