15#ifndef O2_GPUTPCNNCLUSTERIZERHOST_H
16#define O2_GPUTPCNNCLUSTERIZERHOST_H
19#include <unordered_map>
42class GPUReconstruction;
43class GPUTPCNNClusterizer;
44struct GPUSettingsProcessingNNclusterizer;
52 void init(
const GPUSettingsProcessingNNclusterizer&);
A header library for loading ONNX models and inferencing them on CPU and GPU.
void volatileOrtAllocator(Ort::Env *, Ort::MemoryInfo *, GPUReconstruction *, bool=false)
o2::ml::OrtModel mModelReg1
o2::ml::OrtModel mModelClass
const OrtMemoryInfo * getMockedMemoryInfo()
o2::ml::OrtModel mModelReg2
std::unordered_map< std::string, std::string > mOrtOptions
void initClusterizer(const GPUSettingsProcessingNNclusterizer &, GPUTPCNNClusterizer &)
MockedOrtAllocator * getMockedAllocator()
std::shared_ptr< MockedOrtAllocator > mMockedAlloc
std::vector< bool > mModelsUsed
GPUTPCNNClusterizerHost()=default
GPUTPCNNClusterizerHost(const GPUSettingsProcessingNNclusterizer &settings)
void init(const GPUSettingsProcessingNNclusterizer &)