deep_tau::DeepTauCache Class Reference

#include <DeepTauBase.h>

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

Public Member Functions
	DeepTauCache (const std::map< std::string, std::string > &graph_names, bool mem_mapped)
const tensorflow::GraphDef &	getGraph (const std::string &name="") const
tensorflow::Session &	getSession (const std::string &name="") const
	~DeepTauCache ()

Private Attributes
std::map< std::string, GraphPtr >	graphs_
std::map< std::string, std::unique_ptr< tensorflow::MemmappedEnv > >	memmappedEnv_
std::map< std::string, tensorflow::Session * >	sessions_

Detailed Description

Definition at line 48 of file DeepTauBase.h.

Member Typedef Documentation

GraphPtr

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

Definition at line 50 of file DeepTauBase.h.

Constructor & Destructor Documentation

DeepTauCache()

deep_tau::DeepTauCache::DeepTauCache	(	const std::map< std::string, std::string > &	graph_names,
		bool	mem_mapped
	)

Definition at line 210 of file DeepTauBase.cc.

                                                                                             {
    for (const auto& graph_entry : graph_names) {
      tensorflow::SessionOptions options;
      tensorflow::setThreading(options, 1);
 
      const std::string& entry_name = graph_entry.first;
      const std::string& graph_file = graph_entry.second;
      if (mem_mapped) {
        memmappedEnv_[entry_name] = std::make_unique<tensorflow::MemmappedEnv>(tensorflow::Env::Default());
        const tensorflow::Status mmap_status = memmappedEnv_.at(entry_name)->InitializeFromFile(graph_file);
        if (!mmap_status.ok()) {
          throw cms::Exception("DeepTauCache: unable to initalize memmapped environment for ")
              << graph_file << ". \n"
              << mmap_status.ToString();
        }
 
        graphs_[entry_name] = std::make_unique<tensorflow::GraphDef>();
        const tensorflow::Status load_graph_status =
            ReadBinaryProto(memmappedEnv_.at(entry_name).get(),
                            tensorflow::MemmappedFileSystem::kMemmappedPackageDefaultGraphDef,
                            graphs_.at(entry_name).get());
        if (!load_graph_status.ok())
          throw cms::Exception("DeepTauCache: unable to load graph from ") << graph_file << ". \n"
                                                                           << load_graph_status.ToString();
 
        options.config.mutable_graph_options()->mutable_optimizer_options()->set_opt_level(
            ::tensorflow::OptimizerOptions::L0);
        options.env = memmappedEnv_.at(entry_name).get();
 
        sessions_[entry_name] = tensorflow::createSession(graphs_.at(entry_name).get(), options);
 
      } else {
        graphs_[entry_name].reset(tensorflow::loadGraphDef(graph_file));
        sessions_[entry_name] = tensorflow::createSession(graphs_.at(entry_name).get(), options);
      }
    }
  }

References tensorflow::createSession(), Exception, graphs_, tensorflow::loadGraphDef(), memmappedEnv_, AlcaSiPixelAliHarvester0T_cff::options, sessions_, tensorflow::setThreading(), btagGenBb_cfi::Status, and AlCaHLTBitMon_QueryRunRegistry::string.

~DeepTauCache()

deep_tau::DeepTauCache::~DeepTauCache

(

)

Definition at line 248 of file DeepTauBase.cc.

                              {
    for (auto& session_entry : sessions_)
      tensorflow::closeSession(session_entry.second);
  }

References tensorflow::closeSession(), and sessions_.

Member Function Documentation

getGraph()

const tensorflow::GraphDef& deep_tau::DeepTauCache::getGraph ( const std::string &  name = "" ) const

inline

Definition at line 58 of file DeepTauBase.h.

{ return *graphs_.at(name); }

References graphs_, and Skims_PA_cff::name.

Referenced by DeepTauId::DeepTauId(), and DPFIsolation::DPFIsolation().

getSession()

tensorflow::Session& deep_tau::DeepTauCache::getSession ( const std::string &  name = "" ) const

inline

Definition at line 57 of file DeepTauBase.h.

{ return *sessions_.at(name); }

References Skims_PA_cff::name, and sessions_.

Referenced by DeepTauId::getPartialPredictions(), DPFIsolation::getPredictions(), DeepTauId::getPredictionsV1(), and DeepTauId::getPredictionsV2().

Member Data Documentation

graphs_

std::map<std::string, GraphPtr> deep_tau::DeepTauCache::graphs_

private

Definition at line 61 of file DeepTauBase.h.

Referenced by DeepTauCache(), and getGraph().

memmappedEnv_

std::map<std::string, std::unique_ptr<tensorflow::MemmappedEnv> > deep_tau::DeepTauCache::memmappedEnv_

private

Definition at line 63 of file DeepTauBase.h.

Referenced by DeepTauCache().

sessions_

std::map<std::string, tensorflow::Session*> deep_tau::DeepTauCache::sessions_

private

Definition at line 62 of file DeepTauBase.h.

Referenced by DeepTauCache(), getSession(), and ~DeepTauCache().