StructInfo of Tensor.
More...
#include <struct_info.h>
|
Optional< Expr > | shape |
| optionally store the shape expression of the tensor. More...
|
|
Optional< VDevice > | vdevice |
| The virtual device, indicates where the tensor is expected to be executed. More...
|
|
DataType | dtype |
| The content data type, use void to denote the dtype is unknown. More...
|
|
int | ndim |
| The number of dimension of the tensor, can be unknown. More...
|
|
Span | span |
| Span that points to the original source code. Reserved debug information. More...
|
|
◆ GetShape()
Optional<Array<PrimExpr> > tvm::relax::TensorStructInfoNode::GetShape |
( |
| ) |
const |
|
inline |
- Returns
- Shape if it is known.
◆ IsUnknownDtype()
bool tvm::relax::TensorStructInfoNode::IsUnknownDtype |
( |
| ) |
const |
|
inline |
- Returns
- Whether the struct info contains unknown dtype.
◆ IsUnknownNdim()
bool tvm::relax::TensorStructInfoNode::IsUnknownNdim |
( |
| ) |
const |
|
inline |
- Returns
- Whether the struct info contains unknown ndim.
◆ RegisterReflection()
static void tvm::relax::TensorStructInfoNode::RegisterReflection |
( |
| ) |
|
|
inlinestatic |
◆ TVM_DECLARE_FINAL_OBJECT_INFO()
◆ _type_key
constexpr const char* tvm::relax::TensorStructInfoNode::_type_key = "relax.TensorStructInfo" |
|
staticconstexpr |
◆ dtype
DataType tvm::relax::TensorStructInfoNode::dtype |
The content data type, use void to denote the dtype is unknown.
◆ ndim
int tvm::relax::TensorStructInfoNode::ndim |
The number of dimension of the tensor, can be unknown.
- See also
- kUnknownNDim
◆ shape
Optional<Expr> tvm::relax::TensorStructInfoNode::shape |
optionally store the shape expression of the tensor.
- Note
- shape must be normalized: it can only be std::nullopt or ShapeExpr or Var.
◆ vdevice
Optional<VDevice> tvm::relax::TensorStructInfoNode::vdevice |
The virtual device, indicates where the tensor is expected to be executed.
The documentation for this class was generated from the following file: