Support.hpp File Reference

#include <tensorflow/core/framework/graph.pb.h>
#include <iostream>
#include <string>

Go to the source code of this file.

Data Structures
class	tfkit::CmdArguments
class	tfkit::IOConfiguration

Namespaces
namespace	tfkit
namespace	tfkit::tf

Functions
bool	tfkit::tf::HasAttr (const tensorflow::NodeDef &node, const std::string &attr_name)
tensorflow::DataType	tfkit::tf::GetDataTypeAttr (const tensorflow::NodeDef &node, const std::string &attr_name)
tensorflow::TensorProto *	tfkit::tf::GetTensorAttr (tensorflow::NodeDef &node, const std::string &attr_name)
int	tfkit::tf::GetElementCount (const tensorflow::TensorShapeProto &)
	GetElementCount returns -1 for rank-0 tensor shape.
std::unique_ptr< IOConfiguration >	tfkit::make_ioconfig (const CmdArguments &cmdargs)