ONNXRuntime/interface/tensor_fillers.h

0001 #ifndef RecoBTag_ONNXRuntime_tensor_fillers_h
0002 #define RecoBTag_ONNXRuntime_tensor_fillers_h
0003
0004 #include "DataFormats/BTauReco/interface/DeepFlavourTagInfo.h"
0005
0006 namespace btagbtvdeep {
0007
0008   void jet_tensor_filler(float*& ptr, const btagbtvdeep::DeepFlavourFeatures& features);
0009
0010   void cpf_tensor_filler(float*& ptr, const btagbtvdeep::ChargedCandidateFeatures& c_pf_features);
0011
0012   void npf_tensor_filler(float*& ptr, const btagbtvdeep::NeutralCandidateFeatures& n_pf_features);
0013
0014   void sv_tensor_filler(float*& ptr, const btagbtvdeep::SecondaryVertexFeatures& sv_features);
0015
0016   void jet4vec_tensor_filler(float*& ptr, const btagbtvdeep::JetFeatures& jet_features);
0017
0018   void seedTrack_tensor_filler(float*& ptr, const btagbtvdeep::SeedingTrackFeatures& seed_features);
0019
0020   void neighbourTrack_tensor_filler(float*& ptr, const btagbtvdeep::TrackPairFeatures& neighbourTrack_features);
0021
0022 }  // namespace btagbtvdeep
0023
0024 #endif