TICL/interface/TracksterInferenceByANN.h

0001 #ifndef RecoHGCal_TICL_TracksterInferenceByANN_H__
0002 #define RecoHGCal_TICL_TracksterInferenceByANN_H__
0003
0004 #include "RecoHGCal/TICL/interface/TracksterInferenceAlgoBase.h"
0005
0006 namespace ticl {
0007   class TracksterInferenceByANN : public TracksterInferenceAlgoBase {
0008   public:
0009     explicit TracksterInferenceByANN(const edm::ParameterSet& conf);
0010     void inputData(const std::vector<reco::CaloCluster>& layerClusters, std::vector<Trackster>& tracksters) override;
0011     void runInference(std::vector<Trackster>& tracksters) override;
0012
0013   private:
0014     const cms::Ort::ONNXRuntime* onnxPIDSession_;
0015     const cms::Ort::ONNXRuntime* onnxEnergySession_;
0016   };
0017 }  // namespace ticl
0018
0019 #endif