dlpack-inl.h File Reference

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

Go to the source code of this file.

Namespaces
	tc
	tc::dlutils

Functions
std::string	tc::dlutils::toString (const DLDataType &t)
DLContext	tc::dlutils::getCPUDLContext ()
DLContext	tc::dlutils::getGPUDLContext (int device_id)
template<typename T >
DLDataType	tc::dlutils::getDLDataType ()
template<>
DLDataType	tc::dlutils::getDLDataType< float > ()
template<>
DLDataType	tc::dlutils::getDLDataType< int > ()
void	tc::dlutils::SetSizes (DLTensor &t, const std::vector< int64_t > &sizes)
void	tc::dlutils::SetStrides (DLTensor &t, const std::vector< int64_t > &strides)
void	tc::dlutils::SetStridesFromSizes (DLTensor &t, const std::vector< int64_t > &)
DLTensorUPtr	tc::dlutils::makeDLTensorWithSizes (DLContext ctx, DLDataType dtype, const std::vector< int64_t > &sizes)
std::vector< const DLTensor * >	tc::dlutils::extractRawPtrs (const std::vector< DLTensorUPtr > &uptrs)
std::vector< const DLTensor * >	tc::dlutils::constPtrs (const std::vector< DLTensor * > &ptrs)
DLTensorUPtr	tc::dlutils::makeDLTensor (const DLTensor *ptr)
template<typename T >
std::vector< DLTensorUPtr >	tc::dlutils::makeDLTensorVector (const std::vector< T * > &ptrs)
std::string	tc::dlutils::toString (const DLTensor &t)
bool	tc::dlutils::operator== (const DLDataType &t1, const DLDataType &t2)
std::ostream &	tc::dlutils::operator<< (std::ostream &os, const DLDataType &t)
bool	tc::dlutils::compareDLTensorMetadata (const DLTensor &t1, const DLTensor &t2)
template<typename T , typename TT >
bool	tc::dlutils::compareDLTensorVectorMetadata (const std::vector< T * > &v1, const std::vector< TT * > &v2)