Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2023-03-17 11:16:48

0001 #include "PhysicsTools/TensorFlow/interface/TfGraphDefWrapper.h"
0002 
0003 TfGraphDefWrapper::TfGraphDefWrapper(tensorflow::Session* session, tensorflow::GraphDef* graph)
0004     : session_(session), graph_(graph) {}
0005 const tensorflow::Session* TfGraphDefWrapper::getSession() const { return session_; }
0006 
0007 TfGraphDefWrapper::~TfGraphDefWrapper() { tensorflow::closeSession(session_); };