LLVM: llvm::TensorSpec Class Reference (original) (raw)

#include "[llvm/Analysis/TensorSpec.h](TensorSpec%5F8h%5Fsource.html)"

Public Member Functions
const std::string & name () const
int port () const
TensorType type () const
const std::vector< int64_t > & shape () const
bool operator== (const TensorSpec &Other) const
bool operator!= (const TensorSpec &Other) const
size_t getElementCount () const
Get the number of elements in a tensor with this shape.
size_t getElementByteSize () const
Get the size, in bytes, of one element.
size_t getTotalTensorBufferSize () const
Get the total size of a memory buffer needed to store the whole tensor.
template<typename T >
bool isElementType () const
TensorSpec (const std::string &NewName, const TensorSpec &Other)
void toJSON (json::OStream &OS) const
Static Public Member Functions
template<typename T >
static TensorSpec createSpec (const std::string &Name, const std::vector< int64_t > &Shape, int Port=0)

Definition at line 62 of file TensorSpec.h.

◆ createSpec()

template<typename T >

static TensorSpec llvm::TensorSpec::createSpec ( const std::string & Name, const std::vector< int64_t > & Shape, int Port = 0 ) inlinestatic

◆ getElementByteSize()

size_t llvm::TensorSpec::getElementByteSize ( ) const inline

Get the size, in bytes, of one element.

Definition at line 86 of file TensorSpec.h.

◆ getElementCount()

size_t llvm::TensorSpec::getElementCount ( ) const inline

Get the number of elements in a tensor with this shape.

Definition at line 84 of file TensorSpec.h.

◆ getTotalTensorBufferSize()

size_t llvm::TensorSpec::getTotalTensorBufferSize ( ) const inline

Get the total size of a memory buffer needed to store the whole tensor.

Definition at line 88 of file TensorSpec.h.

◆ isElementType()

template<typename T >

bool llvm::TensorSpec::isElementType ( ) const inline

◆ name()

const std::string & llvm::TensorSpec::name ( ) const inline

◆ operator!=()

◆ operator==()

◆ port()

int llvm::TensorSpec::port ( ) const inline

◆ shape()

const std::vector< int64_t > & llvm::TensorSpec::shape ( ) const inline

◆ toJSON()

◆ type()

TensorType llvm::TensorSpec::type ( ) const inline

The documentation for this class was generated from the following files: