tensor_fillers.h

Go to the documentation of this file.

#ifndef RecoBTag_ONNXRuntime_tensor_fillers_h
#define RecoBTag_ONNXRuntime_tensor_fillers_h
 
#include "DataFormats/BTauReco/interface/DeepFlavourTagInfo.h"
 
namespace btagbtvdeep {
 
  void jet_tensor_filler(float*& ptr, const btagbtvdeep::DeepFlavourFeatures& features);
 
  void cpf_tensor_filler(float*& ptr, const btagbtvdeep::ChargedCandidateFeatures& c_pf_features);
 
  void npf_tensor_filler(float*& ptr, const btagbtvdeep::NeutralCandidateFeatures& n_pf_features);
 
  void sv_tensor_filler(float*& ptr, const btagbtvdeep::SecondaryVertexFeatures& sv_features);
 
  void jet4vec_tensor_filler(float*& ptr, const btagbtvdeep::JetFeatures& jet_features);
 
  void seedTrack_tensor_filler(float*& ptr, const btagbtvdeep::SeedingTrackFeatures& seed_features);
 
  void neighbourTrack_tensor_filler(float*& ptr, const btagbtvdeep::TrackPairFeatures& neighbourTrack_features);
 
}  // namespace btagbtvdeep
 
#endif