TensorFlow.cc File Reference

#include "PhysicsTools/TensorFlow/interface/TensorFlow.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/Utilities/interface/ResourceInformation.h"

Go to the source code of this file.

Namespaces
	tensorflow

Functions
bool	tensorflow::closeSession (Session *&session)
bool	tensorflow::closeSession (const Session *&session)
Session *	tensorflow::createSession (SessionOptions &sessionOptions)
Session *	tensorflow::createSession (Backend backend=Backend::cpu, int nThreads=1)
Session *	tensorflow::createSession (const MetaGraphDef *metaGraphDef, const std::string &exportDir, SessionOptions &sessionOptions)
Session *	tensorflow::createSession (const MetaGraphDef *metaGraphDef, const std::string &exportDir, Backend backend=Backend::cpu, int nThreads=1)
Session *	tensorflow::createSession (const GraphDef *graphDef, SessionOptions &sessionOptions)
Session *	tensorflow::createSession (const GraphDef *graphDef, Backend backend=Backend::cpu, 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, Backend backend=Backend::cpu, int nThreads=1)
MetaGraphDef *	tensorflow::loadMetaGraphDef (const std::string &exportDir, const std::string &tag, SessionOptions &sessionOptions)
MetaGraphDef *	tensorflow::loadMetaGraphDef (const std::string &exportDir, const std::string &tag=kSavedModelTagServe, Backend backend=Backend::cpu, int nThreads=1)
void	tensorflow::run (Session *session, const NamedTensorList &inputs, const std::vector< std::string > &outputNames, std::vector< Tensor > *outputs, const thread::ThreadPoolOptions &threadPoolOptions)
void	tensorflow::run (Session *session, const NamedTensorList &inputs, const std::vector< std::string > &outputNames, std::vector< Tensor > *outputs, thread::ThreadPoolInterface *threadPool)
void	tensorflow::run (Session *session, const NamedTensorList &inputs, const std::vector< std::string > &outputNames, std::vector< Tensor > *outputs, const std::string &threadPoolName="no_threads")
void	tensorflow::run (Session *session, const std::vector< std::string > &outputNames, std::vector< Tensor > *outputs, const std::string &threadPoolName="no_threads")
void	tensorflow::setBackend (SessionOptions &sessionOptions, Backend backend=Backend::cpu)
void	tensorflow::setLogging (const std::string &level="3")
void	tensorflow::setThreading (SessionOptions &sessionOptions, int nThreads=1)
void	tensorflow::setThreading (SessionOptions &sessionOptions, int nThreads, const std::string &singleThreadPool)