CMS 3D CMS Logo

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() [1/3]

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::~TfGraphDefWrapper ( )

Definition at line 7 of file TfGraphDefWrapper.cc.

References tensorflow::closeSession(), and session_.

bool closeSession(Session *&session)
Definition: TensorFlow.cc:233
tensorflow::Session * session_

◆ TfGraphDefWrapper() [2/3]

TfGraphDefWrapper::TfGraphDefWrapper ( const TfGraphDefWrapper )
delete

◆ TfGraphDefWrapper() [3/3]

TfGraphDefWrapper::TfGraphDefWrapper ( TfGraphDefWrapper &&  )
delete

Member Function Documentation

◆ getSession()

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

Definition at line 5 of file TfGraphDefWrapper.cc.

References session_.

5 { return session_; }
tensorflow::Session * session_

◆ operator=() [1/2]

TfGraphDefWrapper& TfGraphDefWrapper::operator= ( const TfGraphDefWrapper )
delete

◆ operator=() [2/2]

TfGraphDefWrapper& TfGraphDefWrapper::operator= ( TfGraphDefWrapper &&  )
delete

Member Data Documentation

◆ graph_

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

Definition at line 18 of file TfGraphDefWrapper.h.

◆ session_

tensorflow::Session* TfGraphDefWrapper::session_
private

Definition at line 17 of file TfGraphDefWrapper.h.

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