TensorFlow.h File Reference

#include "tensorflow/core/framework/tensor.h"
#include "tensorflow/core/lib/core/threadpool.h"
#include "tensorflow/core/lib/io/path.h"
#include "tensorflow/core/public/session.h"
#include "tensorflow/core/util/tensor_bundle/naming.h"
#include "tensorflow/cc/client/client_session.h"
#include "tensorflow/cc/saved_model/loader.h"
#include "tensorflow/cc/saved_model/constants.h"
#include "tensorflow/cc/saved_model/tag_constants.h"
#include "PhysicsTools/TensorFlow/interface/NoThreadPool.h"
#include "PhysicsTools/TensorFlow/interface/TBBThreadPool.h"
#include "FWCore/Utilities/interface/Exception.h"

Go to the source code of this file.

Classes
struct	tensorflow::Options
struct	tensorflow::SessionCache

Namespaces
	tensorflow

Typedefs
typedef std::pair< std::string, Tensor >	tensorflow::NamedTensor
typedef std::vector< NamedTensor >	tensorflow::NamedTensorList

Enumerations
enum	tensorflow::Backend {   tensorflow::Backend::cpu, tensorflow::Backend::cuda, tensorflow::Backend::rocm, tensorflow::Backend::intel,   tensorflow::Backend::best }

Functions
bool	tensorflow::checkEmptyInputs (const NamedTensorList &inputs)
bool	tensorflow::closeSession (Session *&session)
bool	tensorflow::closeSession (const Session *&session)
Session *	tensorflow::createSession ()
Session *	tensorflow::createSession (Options &options)
Session *	tensorflow::createSession (const MetaGraphDef *metaGraphDef, const std::string &exportDir, Options &options)
Session *	tensorflow::createSession (const GraphDef *graphDef)
Session *	tensorflow::createSession (const GraphDef *graphDef, Options &options)
GraphDef *	tensorflow::loadGraphDef (const std::string &pbFile)
MetaGraphDef *	tensorflow::loadMetaGraph (const std::string &exportDir, const std::string &tag, Options &Options)
MetaGraphDef *	tensorflow::loadMetaGraphDef (const std::string &exportDir, const std::string &tag=kSavedModelTagServe)
MetaGraphDef *	tensorflow::loadMetaGraphDef (const std::string &exportDir, const std::string &tag, Options &options)
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 (const 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 (const 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 (const 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::run (const Session *session, const std::vector< std::string > &outputNames, std::vector< Tensor > *outputs, const std::string &threadPoolName="no_threads")
void	tensorflow::setLogging (const std::string &level="3")