Convert.cpp File Reference

#include <plier/tf/Convert.h>
#include <nncc/core/ADT/tensor/Shape.h>
#include <cassert>
#include <stdexcept>

Go to the source code of this file.

Namespaces
namespace	plier
namespace	plier::tf

Functions
bool	plier::tf::has_attr (const tensorflow::NodeDef &node, const std::string &attr_name)
bool	plier::tf::has_attrs (const tensorflow::NodeDef &node, const std::vector< std::string > &attr_names)
tensorflow::DataType	plier::tf::get_datatype_attr (const tensorflow::NodeDef &node, const std::string &attr_name)
const tensorflow::TensorShapeProto &	plier::tf::get_shape_attr (const tensorflow::NodeDef &node, const std::string &attr_name)
const tensorflow::TensorProto &	plier::tf::get_tensor_attr (const tensorflow::NodeDef &node, const std::string &attr_name)
const tensorflow::AttrValue_ListValue &	plier::tf::get_list_attr (const tensorflow::NodeDef &node, const std::string &attr_name)
const std::string &	plier::tf::get_string_attr (const tensorflow::NodeDef &node, const std::string &attr_name)
int64_t	plier::tf::get_int_attr (const tensorflow::NodeDef &node, const std::string &attr_name)
float	plier::tf::get_float_attr (const tensorflow::NodeDef &node, const std::string &attr_name)
bool	plier::tf::get_bool_attr (const tensorflow::NodeDef &node, const std::string &attr_name)
std::vector< int64_t >	plier::tf::as_int64_list (const tensorflow::AttrValue_ListValue &lv)
loco::DataType	plier::tf::as_loco_datatype (const tensorflow::DataType dtype)
DataLayout	plier::tf::as_data_layout (const std::string &tf_layout_str)
	@ brief Convert TF Data Layout string (e.g., "NHWC") to enum class for programming convenience
DataLayout	plier::tf::get_data_layout (const tensorflow::NodeDef &node, const std::string &attr_name)
void	plier::tf::copy_shape (const tensorflow::TensorShapeProto &tf_shape, nncc::core::ADT::tensor::Shape &to_shape)
	Copy shape defined in TensorShapeProto to angkor shape.