tensorflow::SessionCache Struct Reference

#include <TensorFlow.h>

Public Member Functions
void	closeSession ()
template<typename... Args>
void	createSession (const std::string &graphPath, Args &&... sessionArgs)
const Session *	getSession () const
	SessionCache ()
template<typename... Args>
	SessionCache (const std::string &graphPath, Args &&... sessionArgs)
	~SessionCache ()

Public Attributes
std::atomic< GraphDef * >	graph
std::atomic< Session * >	session

Detailed Description

Definition at line 190 of file TensorFlow.h.

Constructor & Destructor Documentation

SessionCache() [1/2]

tensorflow::SessionCache::SessionCache ( )

inline

Definition at line 195 of file TensorFlow.h.

{}

SessionCache() [2/2]

template<typename... Args>

tensorflow::SessionCache::SessionCache ( const std::string &  graphPath, Args &&...  sessionArgs  )

inline

Definition at line 199 of file TensorFlow.h.

References createSession(), and HLT_2024v10_cff::graphPath.

                                                                    {
      createSession(graphPath, std::forward<Args>(sessionArgs)...);
    }

~SessionCache()

tensorflow::SessionCache::~SessionCache ( )

inline

Definition at line 204 of file TensorFlow.h.

References closeSession().

{ closeSession(); }

Member Function Documentation

closeSession()

void tensorflow::SessionCache::closeSession

(

)

Definition at line 318 of file TensorFlow.cc.

References tensorflow::closeSession(), graph, alignCSCRings::s, and session.

Referenced by ~SessionCache().

                                  {
    // delete the session if set
    Session* s = session.load();
    if (s != nullptr) {
      tensorflow::closeSession(s);
      session.store(nullptr);
    }

    // delete the graph if set
    if (graph.load() != nullptr) {
      delete graph.load();
      graph.store(nullptr);
    }
  }

createSession()

template<typename... Args>

void tensorflow::SessionCache::createSession ( const std::string &  graphPath, Args &&...  sessionArgs  )

inline

Definition at line 209 of file TensorFlow.h.

References tensorflow::createSession(), graph, HLT_2024v10_cff::graphPath, tensorflow::loadGraphDef(), and session.

Referenced by SessionCache().

                                                                          {
      graph.store(loadGraphDef(graphPath));
      session.store(tensorflow::createSession(graph.load(), std::forward<Args>(sessionArgs)...));
    }

getSession()

const Session* tensorflow::SessionCache::getSession ( ) const

inline

Definition at line 215 of file TensorFlow.h.

References session.

{ return session.load(); }

Member Data Documentation

graph

std::atomic<GraphDef*> tensorflow::SessionCache::graph

Definition at line 191 of file TensorFlow.h.

Referenced by closeSession(), createSession(), and cuy.FindIssue::startElement().

session

std::atomic<Session*> tensorflow::SessionCache::session

Definition at line 192 of file TensorFlow.h.

Referenced by closeSession(), createSession(), and getSession().