Declared in <llvm/Analysis/TensorSpec.h>
class TensorSpec final| Name | Description |
|---|---|
TensorSpec [constructor] | |
getElementByteSize | Get the size, in bytes, of one element. |
getElementCount | Get the number of elements in a tensor with this shape. |
getTotalTensorBufferSize | Get the total size of a memory buffer needed to store the whole tensor. |
isElementType | |
name | |
port | |
shape | |
toJSON | |
type | |
operator== | Equality operator |
operator!= | Inequality operator |
| Name |
|---|
createSpec |
| Name | Description |
|---|---|
getTensorSpecFromJSON | Construct a TensorSpec from a JSON dictionary of the form: { "name": <string>, "port": <int>, "type": <string. Use LLVM's types, e.g. float, double, int64_t>, "shape": <array of ints> } For the "type" field, see the C++ primitive types used in TFUTILS_SUPPORTED_TYPES. |