Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2022-05-26 03:56:12

0001 #ifndef PhysicsTools_TensorFlow_TfGraphDefWrapper_h
0002 #define PhysicsTools_TensorFlow_TfGraphDefWrapper_h
0003 
0004 #include "PhysicsTools/TensorFlow/interface/TensorFlow.h"
0005 
0006 class TfGraphDefWrapper {
0007 public:
0008   TfGraphDefWrapper(tensorflow::Session*, tensorflow::GraphDef*);
0009   ~TfGraphDefWrapper();
0010   TfGraphDefWrapper(const TfGraphDefWrapper&) = delete;
0011   TfGraphDefWrapper& operator=(const TfGraphDefWrapper&) = delete;
0012   TfGraphDefWrapper(TfGraphDefWrapper&&) = delete;
0013   TfGraphDefWrapper& operator=(TfGraphDefWrapper&&) = delete;
0014   const tensorflow::Session* getSession() const;
0015 
0016 private:
0017   tensorflow::Session* session_;
0018   std::unique_ptr<tensorflow::GraphDef> graph_;
0019 };
0020 
0021 #endif