tc::detail::TensorInfo Struct Reference

#include <compilation_cache.h>

Collaboration diagram for tc::detail::TensorInfo:

Public Member Functions
	TensorInfo (const DLTensor *t)
	TensorInfo (const TensorInfoProto &buf)
bool	operator== (const DLTensor *t) const
bool	operator== (const TensorInfo &t) const
bool	operator< (const TensorInfo &t) const
TensorInfoProto	toProtobuf () const

Public Attributes
std::vector< int64_t >	shape
std::vector< int64_t >	strides
uint64_t	alignment
DLDataType	dType

Detailed Description

TensorInfo wraps the necessary bits of DLTensor that are used as part of the CompilationCache's entry keys.

It is serializable to protobuf and stored directly in the cache.

Constructor & Destructor Documentation

tc::detail::TensorInfo::TensorInfo

(

const DLTensor *

t

)

tc::detail::TensorInfo::TensorInfo

(

const TensorInfoProto &

buf

)

Member Function Documentation

bool tc::detail::TensorInfo::operator<

(

const TensorInfo &

t

)

const

bool tc::detail::TensorInfo::operator==

(

const DLTensor *

t

)

const

bool tc::detail::TensorInfo::operator==

(

const TensorInfo &

t

)

const

TensorInfoProto tc::detail::TensorInfo::toProtobuf

(

)

const

Member Data Documentation

uint64_t tc::detail::TensorInfo::alignment

DLDataType tc::detail::TensorInfo::dType

std::vector<int64_t> tc::detail::TensorInfo::shape

std::vector<int64_t> tc::detail::TensorInfo::strides

---

The documentation for this struct was generated from the following file:

• include/tc/core/compilation_cache.h