PhysicsTools/TensorFlow/src/TfGraphDefWrapper.cc - master - githubmirror/cmssw

PhysicsTools/TensorFlow/src/TfGraphDefWrapper.cc

7 строк372 B

mmusich

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

17 май 2022, 10:21

17 май 2022, 10:21929e238

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

TfGraphDefWrapper::TfGraphDefWrapper(tensorflow::Session* session, tensorflow::GraphDef* graph)
    : session_(session), graph_(graph) {}
const tensorflow::Session* TfGraphDefWrapper::getSession() const { return session_; }

TfGraphDefWrapper::~TfGraphDefWrapper() { tensorflow::closeSession(session_); };