deep_tau::DeepTauCache Class Reference

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 15 of file DeepTauId.cc.

Member Typedef Documentation

GraphPtr

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

Definition at line 17 of file DeepTauId.cc.

Constructor & Destructor Documentation

DeepTauCache()

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

inline

Definition at line 19 of file DeepTauId.cc.

References tensorflow::cpu, tensorflow::createSession(), cms::soa::RestrictQualify::Default, Exception, HLT_2024v14_cff::graph_file, graphs_, tensorflow::loadGraphDef(), HLT_2024v14_cff::mem_mapped, memmappedEnv_, AlcaSiPixelAliHarvester0T_cff::options, sessions_, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                   {
      for (const auto& graph_entry : graph_names) {
        // Backend : to be parametrized from the python config
        tensorflow::Options options{tensorflow::Backend::cpu};

        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.getSessionOptions().config.mutable_graph_options()->mutable_optimizer_options()->set_opt_level(
              ::tensorflow::OptimizerOptions::L0);
          options.getSessionOptions().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);
        }
      }
    };

~DeepTauCache()

deep_tau::DeepTauCache::~DeepTauCache ( )

inline

Definition at line 56 of file DeepTauId.cc.

References tensorflow::closeSession(), and sessions_.

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

Member Function Documentation

getGraph()

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

inline

Definition at line 64 of file DeepTauId.cc.

References graphs_, and Skims_PA_cff::name.

{ return *graphs_.at(name); }

getSession()

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

inline

Definition at line 63 of file DeepTauId.cc.

References Skims_PA_cff::name, and sessions_.

Referenced by DeepTauId::getPartialPredictions(), and DeepTauId::getPredictionsV2().

{ return *sessions_.at(name); }

Member Data Documentation

graphs_

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

private

Definition at line 67 of file DeepTauId.cc.

Referenced by DeepTauCache(), and getGraph().

memmappedEnv_

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

private

Definition at line 69 of file DeepTauId.cc.

Referenced by DeepTauCache().

sessions_

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

private

Definition at line 68 of file DeepTauId.cc.

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