PhysicsTools
TensorFlow
src
TfGraphDefWrapper.cc
Go to the documentation of this file.
1
#include "
PhysicsTools/TensorFlow/interface/TfGraphDefWrapper.h
"
2
3
TfGraphDefWrapper::TfGraphDefWrapper
(tensorflow::Session* session, tensorflow::GraphDef* graph)
4
: session_(session), graph_(graph) {}
5
const
tensorflow::Session*
TfGraphDefWrapper::getSession
()
const
{
return
session_
; }
6
7
TfGraphDefWrapper::~TfGraphDefWrapper
() {
tensorflow::closeSession
(
session_
); };
TfGraphDefWrapper::getSession
const tensorflow::Session * getSession() const
Definition:
TfGraphDefWrapper.cc:5
TfGraphDefWrapper.h
TfGraphDefWrapper::~TfGraphDefWrapper
~TfGraphDefWrapper()
Definition:
TfGraphDefWrapper.cc:7
tensorflow::closeSession
bool closeSession(Session *&session)
Definition:
TensorFlow.cc:197
TfGraphDefWrapper::TfGraphDefWrapper
TfGraphDefWrapper(tensorflow::Session *, tensorflow::GraphDef *)
Definition:
TfGraphDefWrapper.cc:3
TfGraphDefWrapper::session_
tensorflow::Session * session_
Definition:
TfGraphDefWrapper.h:17
Generated for CMSSW Reference Manual by
1.8.14