ONNXRuntime/interface/tensor_configs.h

0001 #ifndef RecoBTag_ONNXRuntime_tensor_configs_h
0002 #define RecoBTag_ONNXRuntime_tensor_configs_h
0003
0004 #include <array>
0005 namespace deepflavour {
0006
0007   constexpr unsigned n_features_global = 15;
0008
0009   constexpr unsigned n_cpf = 25;
0010   constexpr unsigned n_features_cpf = 16;
0011
0012   constexpr unsigned n_npf = 25;
0013   constexpr unsigned n_features_npf = 6;
0014
0015   constexpr unsigned n_sv = 4;
0016   constexpr unsigned n_features_sv = 12;
0017
0018 }  // namespace deepflavour
0019
0020 namespace deepvertex {
0021
0022   constexpr unsigned n_features_global = 4;
0023
0024   constexpr unsigned n_seed = 10;
0025   constexpr unsigned n_features_seed = 21;
0026
0027   constexpr unsigned n_neighbor = 20;
0028   constexpr unsigned n_features_neighbor = 36;
0029
0030 }  // namespace deepvertex
0031
0032 namespace parT {
0033
0034   enum InputFeatures {
0035     kBegin = 0,
0036     kChargedCandidates = kBegin,
0037     kNeutralCandidates = 1,
0038     kVertices = 2,
0039     kChargedCandidates4Vec = 3,
0040     kNeutralCandidates4Vec = 4,
0041     kVertices4Vec = 5,
0042     kEnd = 6
0043   };
0044
0045   inline constexpr unsigned n_cpf_accept = 25;
0046   inline constexpr unsigned n_npf_accept = 25;
0047   inline constexpr unsigned n_sv_accept = 5;
0048
0049   constexpr std::array<unsigned int, kEnd> N_InputFeatures{{
0050       16,  // kChargedCandidates
0051       8,   // kNeutralCandidates
0052       14,  // kVertices
0053       4,   // kChargedCandidates4Vec
0054       4,   // kNeutralCandidates4Vec
0055       4,   // kVertices4Vec
0056   }};
0057
0058   constexpr std::array<unsigned int, kEnd> N_AcceptedFeatures{{
0059       n_cpf_accept,  // kChargedCandidates
0060       n_npf_accept,  // kNeutralCandidates
0061       n_sv_accept,   // kVertices
0062       n_cpf_accept,  // kChargedCandidates4Vec
0063       n_npf_accept,  // kNeutralCandidates4Vec
0064       n_sv_accept,   // kVertices4Vec
0065   }};
0066
0067 }  // namespace parT
0068
0069 namespace UparT {
0070
0071   enum InputFeatures {
0072     kBegin = 0,
0073     kChargedCandidates = kBegin,
0074     kLostTracks = 1,
0075     kNeutralCandidates = 2,
0076     kVertices = 3,
0077     kChargedCandidates4Vec = 4,
0078     kLostTracks4Vec = 5,
0079     kNeutralCandidates4Vec = 6,
0080     kVertices4Vec = 7,
0081     kEnd = 8
0082   };
0083
0084   inline constexpr unsigned n_cpf_accept = 29;
0085   inline constexpr unsigned n_lt_accept = 5;
0086   inline constexpr unsigned n_npf_accept = 25;
0087   inline constexpr unsigned n_sv_accept = 5;
0088
0089   constexpr std::array<unsigned int, kEnd> N_InputFeatures{{
0090       25,  // kChargedCandidates
0091       18,  // kLostTracks
0092       8,   // kNeutralCandidates
0093       14,  // kVertices
0094       4,   // kChargedCandidates4Vec
0095       4,   // kLostTracks4Vec
0096       4,   // kNeutralCandidates4Vec
0097       4,   // kVertices4Vec
0098   }};
0099
0100   constexpr std::array<unsigned int, kEnd> N_AcceptedFeatures{{
0101       n_cpf_accept,  // kChargedCandidates
0102       n_lt_accept,   // kLostTracks
0103       n_npf_accept,  // kNeutralCandidates
0104       n_sv_accept,   // kVertices
0105       n_cpf_accept,  // kChargedCandidates4Vec
0106       n_lt_accept,   // kLostTracks4Vec
0107       n_npf_accept,  // kNeutralCandidates4Vec
0108       n_sv_accept,   // kVertices4Vec
0109   }};
0110
0111 }  // namespace UparT
0112 #endif