CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Groups Pages
List of all members | Public Member Functions | Private Attributes
TfGraphDefWrapper Class Reference

#include <TfGraphDefWrapper.h>

Public Member Functions

const tensorflow::Session * getSession () const
 
TfGraphDefWrapperoperator= (const TfGraphDefWrapper &)=delete
 
TfGraphDefWrapperoperator= (TfGraphDefWrapper &&)=delete
 
 TfGraphDefWrapper (tensorflow::Session *, tensorflow::GraphDef *)
 
 TfGraphDefWrapper (const TfGraphDefWrapper &)=delete
 
 TfGraphDefWrapper (TfGraphDefWrapper &&)=delete
 
 ~TfGraphDefWrapper ()
 

Private Attributes

std::unique_ptr
< tensorflow::GraphDef > 
graph_
 
tensorflow::Session * session_
 

Detailed Description

Definition at line 6 of file TfGraphDefWrapper.h.

Constructor & Destructor Documentation

TfGraphDefWrapper::TfGraphDefWrapper ( tensorflow::Session *  session,
tensorflow::GraphDef *  graph 
)

Definition at line 3 of file TfGraphDefWrapper.cc.

4  : session_(session), graph_(graph) {}
tensorflow::Session * session_
std::unique_ptr< tensorflow::GraphDef > graph_
TfGraphDefWrapper::~TfGraphDefWrapper ( )

Definition at line 7 of file TfGraphDefWrapper.cc.

References tensorflow::closeSession(), and session_.

bool closeSession(Session *&session)
Definition: TensorFlow.cc:198
tensorflow::Session * session_
TfGraphDefWrapper::TfGraphDefWrapper ( const TfGraphDefWrapper )
delete
TfGraphDefWrapper::TfGraphDefWrapper ( TfGraphDefWrapper &&  )
delete

Member Function Documentation

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

Definition at line 5 of file TfGraphDefWrapper.cc.

References session_.

5 { return session_; }
tensorflow::Session * session_
TfGraphDefWrapper& TfGraphDefWrapper::operator= ( const TfGraphDefWrapper )
delete
TfGraphDefWrapper& TfGraphDefWrapper::operator= ( TfGraphDefWrapper &&  )
delete

Member Data Documentation

std::unique_ptr<tensorflow::GraphDef> TfGraphDefWrapper::graph_
private

Definition at line 18 of file TfGraphDefWrapper.h.

tensorflow::Session* TfGraphDefWrapper::session_
private

Definition at line 17 of file TfGraphDefWrapper.h.

Referenced by getSession(), and ~TfGraphDefWrapper().