Numpy.hpp File Reference

#include <fmt/format.h>
#include "Types.hpp"
#include "Tensor.hpp"
#include <fstream>
#include <iostream>

Include dependency graph for Numpy.hpp:

Go to the source code of this file.

Classes
struct	HeaderInfo
struct	Header

Namespaces
	armnnNumpy

Functions
void	CreateHeaderInfo (std::ifstream &ifStream, HeaderInfo &headerInfo)
std::string	getSubstring (std::string fullString, std::string substringStart, std::string substringEnd, bool removeStartChar=0, bool includeEndChar=0)
	Primarily used to isolate values from header dictionary. More...
void	parseShape (Header &header, std::string &shapeString)
void	CreateHeader (std::ifstream &ifStream, HeaderInfo &headerInfo, Header &header)
template<typename T >
void	ReadData (std::ifstream &ifStream, T *tensor, const unsigned int &numElements)
armnn::DataType	getArmNNDataType (std::string &descr)
std::string	getNumpyDescr (armnn::DataType dType)
template<typename T >
bool	compareCTypes (std::string &descr)
unsigned int	getNumElements (Header &header)
template<typename T >
void	WriteToNumpyFile (const std::string &outputTensorFileName, const T *const array, const unsigned int numElements, armnn::DataType dataType, const armnn::TensorShape &shape)