BJetTFCache Struct Reference

#include <JetId.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 14 of file JetId.h.

Constructor & Destructor Documentation

BJetTFCache()

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

inline

Definition at line 15 of file JetId.h.

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

                                          : graphDef(tensorflow::loadGraphDef(graphPath)) {
    session = tensorflow::createSession(graphDef.get());
  }

~BJetTFCache()

BJetTFCache::~BJetTFCache ( )

inline

Definition at line 18 of file JetId.h.

References tensorflow::closeSession(), and session.

{ tensorflow::closeSession(session); }

Member Data Documentation

graphDef

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

Definition at line 19 of file JetId.h.

Referenced by BJetTFCache().

session

tensorflow::Session* BJetTFCache::session

Definition at line 20 of file JetId.h.

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