Constant tensor.
More...
#include <expr.h>
|
runtime::NDArray | data |
| The data of the tensor. More...
|
|
Optional< ObjectRef > | struct_info_ = Optional<ObjectRef>() |
| Stores the result of structure information of the expression that encapsulate both static shape and runtime information such as shape. More...
|
|
Span | span |
| Span that points to the original source code. Reserved debug information. More...
|
|
Constant tensor.
- Note
- Scalar constants are represented by ndim-0 constant tensors.
◆ is_scalar()
bool tvm::relax::ConstantNode::is_scalar |
( |
| ) |
const |
|
inline |
- Returns
- Whether it is scalar(ndim-0 tensor)
◆ RegisterReflection()
static void tvm::relax::ConstantNode::RegisterReflection |
( |
| ) |
|
|
inlinestatic |
◆ tensor_type()
TensorType tvm::relax::ConstantNode::tensor_type |
( |
| ) |
const |
- Returns
- The corresponding tensor type of the data
◆ TVM_DECLARE_FINAL_OBJECT_INFO()
◆ _type_key
constexpr const char* tvm::relax::ConstantNode::_type_key = "relax.expr.Constant" |
|
staticconstexpr |
◆ data
The documentation for this class was generated from the following file: