ONNXRuntime/interface/tensor_configs.h

0001 #ifndef RecoBTag_ONNXRuntime_tensor_configs_h
0002 #define RecoBTag_ONNXRuntime_tensor_configs_h
0003
0004 namespace deepflavour {
0005
0006   constexpr unsigned n_features_global = 15;
0007
0008   constexpr unsigned n_cpf = 25;
0009   constexpr unsigned n_features_cpf = 16;
0010
0011   constexpr unsigned n_npf = 25;
0012   constexpr unsigned n_features_npf = 6;
0013
0014   constexpr unsigned n_sv = 4;
0015   constexpr unsigned n_features_sv = 12;
0016
0017 }  // namespace deepflavour
0018
0019 namespace deepvertex {
0020
0021   constexpr unsigned n_features_global = 4;
0022
0023   constexpr unsigned n_seed = 10;
0024   constexpr unsigned n_features_seed = 21;
0025
0026   constexpr unsigned n_neighbor = 20;
0027   constexpr unsigned n_features_neighbor = 36;
0028
0029 }  // namespace deepvertex
0030
0031 #endif