BaseMVACache Class Reference

#include <BaseMVAValueMapProducer.h>

Public Member Functions
	BaseMVACache (const std::string &model_path, const std::string &backend)
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 ( const std::string &  model_path, const std::string &  backend  )

inline

Definition at line 58 of file BaseMVAValueMapProducer.h.

References tensorflow::createSession(), graph_, tensorflow::loadGraphDef(), ort_, and tf_session_.

                                                                      {
    if (backend == "TF") {
      graph_.reset(tensorflow::loadGraphDef(model_path));
      tf_session_ = tensorflow::createSession(graph_.get());
    } else if (backend == "ONNX") {
      ort_ = std::make_unique<cms::Ort::ONNXRuntime>(model_path);
    }
  }

BaseMVACache::~BaseMVACache ( )

inline

Definition at line 66 of file BaseMVAValueMapProducer.h.

References tensorflow::closeSession(), and tf_session_.

{ tensorflow::closeSession(tf_session_); }

Member Function Documentation

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

inline

Definition at line 69 of file BaseMVAValueMapProducer.h.

References ort_.

{ return *ort_; }

tensorflow::Session* BaseMVACache::getTFSession ( ) const

inline

Definition at line 68 of file BaseMVAValueMapProducer.h.

References tf_session_.

{ return tf_session_; }

Member Data Documentation

std::shared_ptr<tensorflow::GraphDef> BaseMVACache::graph_

private

Definition at line 72 of file BaseMVAValueMapProducer.h.

Referenced by BaseMVACache().

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

private

Definition at line 74 of file BaseMVAValueMapProducer.h.

Referenced by BaseMVACache(), and getONNXSession().

tensorflow::Session* BaseMVACache::tf_session_ = nullptr

private

Definition at line 73 of file BaseMVAValueMapProducer.h.

Referenced by BaseMVACache(), getTFSession(), and ~BaseMVACache().