PhysicsTools/TensorFlow/interface/TfGraphDefWrapper.h - master - githubmirror/cmssw

PhysicsTools/TensorFlow/interface/TfGraphDefWrapper.h

21 строка675 B

mmusich

move TfGraphDefWrapper and TfGraphDefProducer from RecoTracker/FinalTrackSelectors to PhysicsTools/TensorFlow

17 май 2022, 10:21

17 май 2022, 10:21929e238

#ifndef PhysicsTools_TensorFlow_TfGraphDefWrapper_h
#define PhysicsTools_TensorFlow_TfGraphDefWrapper_h

#include "PhysicsTools/TensorFlow/interface/TensorFlow.h"

class TfGraphDefWrapper {
public:
  TfGraphDefWrapper(tensorflow::Session*, tensorflow::GraphDef*);
  ~TfGraphDefWrapper();
  TfGraphDefWrapper(const TfGraphDefWrapper&) = delete;
  TfGraphDefWrapper& operator=(const TfGraphDefWrapper&) = delete;
  TfGraphDefWrapper(TfGraphDefWrapper&&) = delete;
  TfGraphDefWrapper& operator=(TfGraphDefWrapper&&) = delete;
  const tensorflow::Session* getSession() const;

private:
  tensorflow::Session* session_;
  std::unique_ptr<tensorflow::GraphDef> graph_;
};

#endif