deep_tau::DeepTauCache Class Reference

#include <DeepTauBase.h>

Public Types
using	GraphPtr = std::shared_ptr< tensorflow::GraphDef >

Public Member Functions
	DeepTauCache (const std::string &graph_name, bool mem_mapped)
const tensorflow::GraphDef &	getGraph () const
tensorflow::Session &	getSession () const
	~DeepTauCache ()

Private Attributes
GraphPtr	graph_
std::unique_ptr< tensorflow::MemmappedEnv >	memmappedEnv_
tensorflow::Session *	session_

Detailed Description

Definition at line 40 of file DeepTauBase.h.

Member Typedef Documentation

using deep_tau::DeepTauCache::GraphPtr = std::shared_ptr<tensorflow::GraphDef>

Definition at line 42 of file DeepTauBase.h.

Constructor & Destructor Documentation

deep_tau::DeepTauCache::DeepTauCache	(	const std::string &	graph_name,
		bool	mem_mapped
	)

Definition at line 120 of file DeepTauBase.cc.

References tensorflow::createSession(), Default, tensorflow::loadGraphDef(), AlcaSiPixelAliHarvester0T_cff::options, tensorflow::setThreading(), and btagGenBb_cfi::Status.

{
    tensorflow::SessionOptions options;
    tensorflow::setThreading(options, 1, "no_threads");

    if(mem_mapped) {
        memmappedEnv_ = std::make_unique<tensorflow::MemmappedEnv>(tensorflow::Env::Default());
        const tensorflow::Status mmap_status = memmappedEnv_.get()->InitializeFromFile(graph_name);
        if(!mmap_status.ok())
            throw cms::Exception("DeepTauCache: unable to initalize memmapped environment for ") << graph_name << ". \n"
                                                                                                 << mmap_status.ToString();

        graph_ = std::make_unique<tensorflow::GraphDef>();
        const tensorflow::Status load_graph_status = ReadBinaryProto(memmappedEnv_.get(),
                                                                     tensorflow::MemmappedFileSystem::kMemmappedPackageDefaultGraphDef,
                                                                     graph_.get());
        if(!load_graph_status.ok())
            throw cms::Exception("DeepTauCache: unable to load graph_ from ") << graph_name << ". \n"
                                                                             << mmap_status.ToString();
        options.config.mutable_graph_options()->mutable_optimizer_options()->set_opt_level(::tensorflow::OptimizerOptions::L0);
        options.env = memmappedEnv_.get();

        session_ = tensorflow::createSession(graph_.get(), options);

    } else {
        graph_.reset(tensorflow::loadGraphDef(graph_name));
        session_ = tensorflow::createSession(graph_.get(), options);
      }
}

deep_tau::DeepTauCache::~DeepTauCache

(

)

Definition at line 150 of file DeepTauBase.cc.

References tensorflow::closeSession().

{
    tensorflow::closeSession(session_);
}

Member Function Documentation

const tensorflow::GraphDef& deep_tau::DeepTauCache::getGraph ( ) const

inline

Definition at line 50 of file DeepTauBase.h.

Referenced by DPFIsolation::DPFIsolation().

{ return *graph_; }

tensorflow::Session& deep_tau::DeepTauCache::getSession ( ) const

inline

Definition at line 49 of file DeepTauBase.h.

Referenced by DPFIsolation::getPredictions().

{ return *session_; }

Member Data Documentation

GraphPtr deep_tau::DeepTauCache::graph_

private

Definition at line 53 of file DeepTauBase.h.

std::unique_ptr<tensorflow::MemmappedEnv> deep_tau::DeepTauCache::memmappedEnv_

private

Definition at line 55 of file DeepTauBase.h.

tensorflow::Session* deep_tau::DeepTauCache::session_

private

Definition at line 54 of file DeepTauBase.h.