CMS 3D CMS Logo

All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
TfGraphDefWrapper.h
Go to the documentation of this file.
1 #ifndef RecoTracker_FinalTrackSelectors_TfGraphDefWrapper_h
2 #define RecoTracker_FinalTrackSelectors_TfGraphDefWrapper_h
3 
5 
7 public:
8  TfGraphDefWrapper(tensorflow::Session*, tensorflow::GraphDef*);
10  TfGraphDefWrapper(const TfGraphDefWrapper&) = delete;
14  const tensorflow::Session* getSession() const;
15 
16 private:
17  tensorflow::Session* session_;
18  std::unique_ptr<tensorflow::GraphDef> graph_;
19 };
20 
21 #endif
TfGraphDefWrapper::getSession
const tensorflow::Session * getSession() const
Definition: TfGraphDefWrapper.cc:5
TensorFlow.h
TfGraphDefWrapper::~TfGraphDefWrapper
~TfGraphDefWrapper()
Definition: TfGraphDefWrapper.cc:7
TfGraphDefWrapper::TfGraphDefWrapper
TfGraphDefWrapper(tensorflow::Session *, tensorflow::GraphDef *)
Definition: TfGraphDefWrapper.cc:3
TfGraphDefWrapper::session_
tensorflow::Session * session_
Definition: TfGraphDefWrapper.h:17
TfGraphDefWrapper
Definition: TfGraphDefWrapper.h:6
TfGraphDefWrapper::graph_
std::unique_ptr< tensorflow::GraphDef > graph_
Definition: TfGraphDefWrapper.h:18
TfGraphDefWrapper::operator=
TfGraphDefWrapper & operator=(const TfGraphDefWrapper &)=delete