data_type.h File Reference

#include <tvm/runtime/c_runtime_api.h>
#include <tvm/runtime/logging.h>
#include <string>
#include <type_traits>

Include dependency graph for data_type.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::runtime::DataType
	Runtime primitive data type. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::runtime

Typedefs
using	tvm::DataType = runtime::DataType

Functions
int	tvm::runtime::GetVectorBytes (DataType dtype)
	Get the number of bytes needed in a vector. More...
bool	tvm::runtime::TypeMatch (DLDataType t, int code, int bits, int lanes=1)
	Check whether type matches the given spec. More...
bool	tvm::runtime::TypeEqual (DLDataType lhs, DLDataType rhs)
	Check whether two types are equal . More...
std::string	tvm::runtime::GetCustomTypeName (uint8_t type_code)
	Runtime utility for getting custom type name from code. More...
bool	tvm::runtime::GetCustomTypeRegistered (uint8_t type_code)
	Runtime utility for checking whether custom type is registered. More...
uint8_t	tvm::runtime::ParseCustomDatatype (const std::string &s, const char **scan)
	Runtime utility for parsing string of the form "custom[<typename>]". More...
const char *	tvm::runtime::DLDataTypeCode2Str (DLDataTypeCode type_code)
	Convert type code to its name. More...
DLDataType	tvm::runtime::String2DLDataType (std::string s)
	convert a string to TVM type. More...
std::string	tvm::runtime::DLDataType2String (DLDataType t)
	convert a TVM type to string. More...
std::ostream &	tvm::runtime::operator<< (std::ostream &os, DLDataType t)
std::ostream &	tvm::runtime::operator<< (std::ostream &os, const DataType &dtype)