CMS 3D CMS Logo

List of all members | Public Member Functions | Public Attributes
BJetTFCache Struct Reference

#include <BJetId.h>

Public Member Functions

 BJetTFCache (const std::string &graphPath)
 
 ~BJetTFCache ()
 

Public Attributes

std::unique_ptr< tensorflow::GraphDef > graphDef
 
tensorflow::Session * session
 

Detailed Description

Definition at line 9 of file BJetId.h.

Constructor & Destructor Documentation

◆ BJetTFCache()

BJetTFCache::BJetTFCache ( const std::string &  graphPath)
inline

Definition at line 10 of file BJetId.h.

References tensorflow::createSession(), graphDef, and session.

12  }
GraphDef * loadGraphDef(const std::string &pbFile)
Definition: TensorFlow.cc:129
std::unique_ptr< tensorflow::GraphDef > graphDef
Definition: BJetId.h:14
Session * createSession()
Definition: TensorFlow.cc:146
tensorflow::Session * session
Definition: BJetId.h:15

◆ ~BJetTFCache()

BJetTFCache::~BJetTFCache ( )
inline

Definition at line 13 of file BJetId.h.

References tensorflow::closeSession(), and session.

bool closeSession(Session *&session)
Definition: TensorFlow.cc:243
tensorflow::Session * session
Definition: BJetId.h:15

Member Data Documentation

◆ graphDef

std::unique_ptr<tensorflow::GraphDef> BJetTFCache::graphDef

Definition at line 14 of file BJetId.h.

Referenced by BJetTFCache().

◆ session

tensorflow::Session* BJetTFCache::session

Definition at line 15 of file BJetId.h.

Referenced by BJetTFCache(), and ~BJetTFCache().