File indexing completed on 2024-04-06 12:24:15
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