CMS 3D CMS Logo

Namespaces | Functions
TensorFlow.cc File Reference
#include "PhysicsTools/TensorFlow/interface/TensorFlow.h"

Go to the source code of this file.

Namespaces

 tensorflow
 

Functions

bool tensorflow::closeSession (Session *&session)
 
Sessiontensorflow::createSession (SessionOptions &sessionOptions)
 
Sessiontensorflow::createSession (int nThreads=1)
 
Sessiontensorflow::createSession (MetaGraphDef *metaGraph, const std::string &exportDir, SessionOptions &sessionOptions)
 
Sessiontensorflow::createSession (MetaGraphDef *metaGraph, const std::string &exportDir, int nThreads=1)
 
Sessiontensorflow::createSession (GraphDef *graphDef, SessionOptions &sessionOptions)
 
Sessiontensorflow::createSession (GraphDef *graphDef, int nThreads=1)
 
GraphDef * tensorflow::loadGraphDef (const std::string &pbFile)
 
MetaGraphDef * tensorflow::loadMetaGraph (const std::string &exportDir, const std::string &tag, SessionOptions &sessionOptions)
 
MetaGraphDef * tensorflow::loadMetaGraph (const std::string &exportDir, const std::string &tag=kSavedModelTagServe, int nThreads=1)
 
void tensorflow::run (Session *session, const NamedTensorList &inputs, const std::vector< std::string > &outputNames, const std::vector< std::string > &targetNodes, std::vector< Tensor > *outputs)
 
void tensorflow::run (Session *session, const std::vector< std::string > &inputNames, const std::vector< Tensor > &inputTensors, const std::vector< std::string > &outputNames, const std::vector< std::string > &targetNodes, std::vector< Tensor > *outputs)
 
void tensorflow::run (Session *session, const NamedTensorList &inputs, const std::vector< std::string > &outputNames, std::vector< Tensor > *outputs)
 
void tensorflow::run (Session *session, const std::vector< std::string > &inputNames, const std::vector< Tensor > &inputTensors, const std::vector< std::string > &outputNames, std::vector< Tensor > *outputs)
 
void tensorflow::setLogging (const std::string &level="3")
 
void tensorflow::setThreading (SessionOptions &sessionOptions, int nThreads, const std::string &singleThreadPool="no_threads")