Support.cpp File Reference

#include "Support.hpp"
#include <tensorflow/core/framework/graph.pb.h>
#include <memory>
#include <cassert>
#include <fstream>
#include <stdexcept>

Go to the source code of this file.

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)