![]() |
![]() |
Public Member Functions | |
DeepCombinedONNXJetTagsProducer (const edm::ParameterSet &, const ONNXRuntime *) | |
~DeepCombinedONNXJetTagsProducer () override | |
![]() | |
EDProducer ()=default | |
bool | hasAbilityToProduceInLumis () const final |
bool | hasAbilityToProduceInRuns () const final |
Static Public Member Functions | |
static void | fillDescriptions (edm::ConfigurationDescriptions &) |
static void | globalEndJob (const ONNXRuntime *) |
static std::unique_ptr< ONNXRuntime > | initializeGlobalCache (const edm::ParameterSet &) |
Private Types | |
enum | InputIndexes { kGlobal = 0, kChargedCandidates = 1, kNeutralCandidates = 2, kVertices = 3, kGlobal1 = 4, kSeedingTracks = 5, kNeighbourTracks = 6 } |
typedef reco::JetTagCollection | JetTagCollection |
typedef std::vector< reco::DeepFlavourTagInfo > | TagInfoCollection |
Private Member Functions | |
void | make_inputs (unsigned i_jet, const reco::DeepFlavourTagInfo &taginfo) |
void | produce (edm::Event &, const edm::EventSetup &) override |
Private Attributes | |
FloatArrays | data_ |
std::vector< std::string > | flav_names_ |
std::vector< std::string > | input_names_ |
const double | max_jet_eta_ |
const double | min_jet_pt_ |
std::vector< std::string > | output_names_ |
const edm::EDGetTokenT< TagInfoCollection > | src_ |
Static Private Attributes | |
static const std::vector< unsigned > | input_sizes_ |
static const unsigned | n_cpf_ = deepflavour::n_cpf |
static const unsigned | n_features_cpf_ = deepflavour::n_features_cpf |
static const unsigned | n_features_global1_ = deepvertex::n_features_global |
static const unsigned | n_features_global_ = deepflavour::n_features_global |
static const unsigned | n_features_neighbor_ = deepvertex::n_features_neighbor |
static const unsigned | n_features_npf_ = deepflavour::n_features_npf |
static const unsigned | n_features_seed_ = deepvertex::n_features_seed |
static const unsigned | n_features_sv_ = deepflavour::n_features_sv |
static const unsigned | n_neighbor_ = deepvertex::n_neighbor |
static const unsigned | n_npf_ = deepflavour::n_npf |
static const unsigned | n_seed_ = deepvertex::n_seed |
static const unsigned | n_sv_ = deepflavour::n_sv |
Additional Inherited Members | |
![]() | |
typedef CacheContexts< T... > | CacheTypes |
typedef CacheTypes::GlobalCache | GlobalCache |
typedef AbilityChecker< T... > | HasAbility |
typedef CacheTypes::LuminosityBlockCache | LuminosityBlockCache |
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache > | LuminosityBlockContext |
typedef CacheTypes::LuminosityBlockSummaryCache | LuminosityBlockSummaryCache |
typedef CacheTypes::RunCache | RunCache |
typedef RunContextT< RunCache, GlobalCache > | RunContext |
typedef CacheTypes::RunSummaryCache | RunSummaryCache |
Definition at line 23 of file DeepCombinedONNXJetTagsProducer.cc.
Definition at line 35 of file DeepCombinedONNXJetTagsProducer.cc.
|
private |
Definition at line 34 of file DeepCombinedONNXJetTagsProducer.cc.
|
private |
Enumerator | |
---|---|
kGlobal | |
kChargedCandidates | |
kNeutralCandidates | |
kVertices | |
kGlobal1 | |
kSeedingTracks | |
kNeighbourTracks |
Definition at line 49 of file DeepCombinedONNXJetTagsProducer.cc.
|
explicit |
Definition at line 94 of file DeepCombinedONNXJetTagsProducer.cc.
References flav_names_, input_names_, and input_sizes_.
|
override |
Definition at line 110 of file DeepCombinedONNXJetTagsProducer.cc.
|
static |
Definition at line 112 of file DeepCombinedONNXJetTagsProducer.cc.
References edm::ConfigurationDescriptions::add(), and edm::ParameterSetDescription::add().
|
static |
Definition at line 147 of file DeepCombinedONNXJetTagsProducer.cc.
|
static |
Definition at line 143 of file DeepCombinedONNXJetTagsProducer.cc.
References edm::ParameterSet::getParameter().
|
private |
Definition at line 220 of file DeepCombinedONNXJetTagsProducer.cc.
References btagbtvdeep::cpf_tensor_filler(), data_, DEFINE_FWK_MODULE, postprocess-scan-build::features, reco::FeaturesTagInfo< Features >::features(), input_sizes_, btagbtvdeep::jet4vec_tensor_filler(), btagbtvdeep::jet_tensor_filler(), kChargedCandidates, kGlobal, kGlobal1, kNeighbourTracks, kNeutralCandidates, kSeedingTracks, kVertices, min(), n_cpf_, n_features_cpf_, n_features_global1_, n_features_global_, n_features_neighbor_, n_features_npf_, n_features_seed_, n_features_sv_, n_neighbor_, n_npf_, n_seed_, n_sv_, btagbtvdeep::neighbourTrack_tensor_filler(), btagbtvdeep::npf_tensor_filler(), PFRecoTauDiscriminationByIsolation_cfi::offset, btagbtvdeep::seedTrack_tensor_filler(), and btagbtvdeep::sv_tensor_filler().
Referenced by produce().
|
overrideprivate |
Definition at line 149 of file DeepCombinedONNXJetTagsProducer.cc.
References data_, flav_names_, edm::Event::getByToken(), mps_fire::i, input_names_, input_sizes_, make_inputs(), edm::makeRefToBaseProdFrom(), max_jet_eta_, min_jet_pt_, eostools::move(), output_names_, PatBasicFWLiteJetAnalyzer_Selector_cfg::outputs, edm::Event::put(), src_, and lumiTag::taginfo.
|
private |
Definition at line 74 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by make_inputs(), and produce().
|
private |
Definition at line 42 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by DeepCombinedONNXJetTagsProducer(), and produce().
|
private |
Definition at line 43 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by DeepCombinedONNXJetTagsProducer(), and produce().
|
staticprivate |
Definition at line 71 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by DeepCombinedONNXJetTagsProducer(), make_inputs(), and produce().
|
private |
Definition at line 47 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by produce().
|
private |
Definition at line 46 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by produce().
|
staticprivate |
Definition at line 59 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by make_inputs().
|
staticprivate |
Definition at line 60 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by make_inputs().
|
staticprivate |
Definition at line 65 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by make_inputs().
|
staticprivate |
Definition at line 58 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by make_inputs().
|
staticprivate |
Definition at line 69 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by make_inputs().
|
staticprivate |
Definition at line 62 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by make_inputs().
|
staticprivate |
Definition at line 67 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by make_inputs().
|
staticprivate |
Definition at line 64 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by make_inputs().
|
staticprivate |
Definition at line 68 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by make_inputs().
|
staticprivate |
Definition at line 61 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by make_inputs().
|
staticprivate |
Definition at line 66 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by make_inputs().
|
staticprivate |
Definition at line 63 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by make_inputs().
|
private |
Definition at line 44 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by produce().
|
private |
Definition at line 41 of file DeepCombinedONNXJetTagsProducer.cc.
Referenced by produce().