File indexing completed on 2023-03-17 11:17:04
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 }
0023
0024 #endif