File OnnxRuntimeBase.hpp¶
-
namespace Acts
Set the Geometry Context PLUGIN.
Set the Calibration Context PLUGIN.
Convenience functions to ease creation of and Acts::InterpolatedMaterialMap and to avoid code duplication.
Set the Mangetic Field Context PLUGIN.
Convenience functions to ease creation of and Acts::InterpolatedBFieldMap and to avoid code duplication.
Currently implemented for the two most common formats: rz and xyz.
-
class OnnxRuntimeBase
- #include </home/docs/checkouts/readthedocs.org/user_builds/acts/checkouts/v19.7.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