graph_utils.h File Reference

Utility functions for FrozenGraphs. More...

#include <algorithm>
#include <sstream>
#include <stdexcept>
#include <string>
#include <vector>
#include <tensorflow/core/platform/env.h>
#include <tensorflow_cpp/utils.h>

Go to the source code of this file.

Namespaces
namespace	tensorflow_cpp
	Namespace for tensorflow_cpp library.

Functions
tf::GraphDef	tensorflow_cpp::loadFrozenGraph (const std::string &file)
	Loads a TensorFlow graph from a frozen graph file.
bool	tensorflow_cpp::loadGraphIntoSession (tf::Session *session, const tf::GraphDef &graph_def)
	Loads a TensorFlow graph into an existing session.
tf::Session *	tensorflow_cpp::loadFrozenGraphIntoNewSession (const std::string &file, const bool allow_growth=true, const double per_process_gpu_memory_fraction=0, const std::string &visible_device_list="")
	Loads a TensorFlow graph from a frozen graph file into a new session.
std::vector< std::string >	tensorflow_cpp::getGraphInputNames (const tf::GraphDef &graph_def)
	Determines the names of all graph input nodes.
std::vector< std::string >	tensorflow_cpp::getGraphOutputNames (const tf::GraphDef &graph_def)
	Determines the names of all graph output nodes.
std::vector< int >	tensorflow_cpp::getGraphNodeShape (const tf::GraphDef &graph_def, const std::string &node_name)
	Determines the shape of a given graph node.
tf::DataType	tensorflow_cpp::getGraphNodeType (const tf::GraphDef &graph_def, const std::string &node_name)
	Determines the datatype of a given graph node.
std::string	tensorflow_cpp::getGraphInfoString (const tf::GraphDef &graph_def)

Detailed Description

Utility functions for FrozenGraphs.

Definition in file graph_utils.h.