#include <BaseMVAValueMapProducer.h>

Public Member Functions
	BaseMVACache (const std::string &model_path, const std::string &backend, const bool disableONNXGraphOpt)

const cms::Ort::ONNXRuntime &	getONNXSession () const

tensorflow::Session *	getTFSession () const

	~BaseMVACache ()

Private Attributes
std::shared_ptr< tensorflow::GraphDef >	graph_

std::unique_ptr< cms::Ort::ONNXRuntime >	ort_

tensorflow::Session *	tf_session_ = nullptr

Detailed Description

Definition at line 56 of file BaseMVAValueMapProducer.h.

Constructor & Destructor Documentation

◆ BaseMVACache()

BaseMVACache::BaseMVACache	(	const std::string &	model_path,
		const std::string &	backend,
		const bool	disableONNXGraphOpt
	)

inline

Definition at line 58 of file BaseMVAValueMapProducer.h.

References jetsAK4_CHS_cff::backend, tensorflow::createSession(), cms::Ort::ONNXRuntime::defaultSessionOptions(), graph_, tensorflow::loadGraphDef(), HLT_2023v12_cff::model_path, ort_, and tf_session_.

                                                                                                       {
     if (backend == "TF") {
       graph_.reset(tensorflow::loadGraphDef(model_path));
       tf_session_ = tensorflow::createSession(graph_.get());
     } else if (backend == "ONNX") {
       if (disableONNXGraphOpt) {
         Ort::SessionOptions sess_opts;
         sess_opts = cms::Ort::ONNXRuntime::defaultSessionOptions();
         sess_opts.SetGraphOptimizationLevel(GraphOptimizationLevel::ORT_DISABLE_ALL);
         ort_ = std::make_unique<cms::Ort::ONNXRuntime>(model_path, &sess_opts);
       } else {
         ort_ = std::make_unique<cms::Ort::ONNXRuntime>(model_path);
       }
     }
   }