tvm
Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tvm::relax::ConstantNode Class Reference

Constant tensor. More...

#include <expr.h>

Inheritance diagram for tvm::relax::ConstantNode:
Collaboration diagram for tvm::relax::ConstantNode:

Public Member Functions

TensorType tensor_type () const
 
bool is_scalar () const
 
 TVM_DECLARE_FINAL_OBJECT_INFO (ConstantNode, LeafExprNode)
 
- Public Member Functions inherited from tvm::relax::LeafExprNode
 TVM_DECLARE_BASE_OBJECT_INFO (LeafExprNode, ExprNode)
 
- Public Member Functions inherited from tvm::RelaxExprNode
 TVM_DECLARE_BASE_OBJECT_INFO (RelaxExprNode, BaseExprNode)
 
- Public Member Functions inherited from tvm::BaseExprNode
 TVM_DECLARE_BASE_OBJECT_INFO (BaseExprNode, Object)
 

Static Public Member Functions

static void RegisterReflection ()
 
- Static Public Member Functions inherited from tvm::RelaxExprNode
static void RegisterReflection ()
 
- Static Public Member Functions inherited from tvm::BaseExprNode
static void RegisterReflection ()
 

Public Attributes

runtime::NDArray data
 The data of the tensor. More...
 
- Public Attributes inherited from tvm::RelaxExprNode
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...
 
- Public Attributes inherited from tvm::BaseExprNode
Span span
 Span that points to the original source code. Reserved debug information. More...
 

Static Public Attributes

static constexpr const char * _type_key = "relax.expr.Constant"
 
- Static Public Attributes inherited from tvm::relax::LeafExprNode
static constexpr const char * _type_key = "relax.expr.LeafExpr"
 
static constexpr const uint32_t _type_child_slots = 7
 
- Static Public Attributes inherited from tvm::RelaxExprNode
static constexpr const char * _type_key = "ir.RelaxExpr"
 
static constexpr const uint32_t _type_child_slots = 22
 
- Static Public Attributes inherited from tvm::BaseExprNode
static constexpr const char * _type_key = "ir.BaseExpr"
 
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode
 
static constexpr const uint32_t _type_child_slots = 64
 

Detailed Description

Constant tensor.

Note
Scalar constants are represented by ndim-0 constant tensors.

Member Function Documentation

◆ 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()

tvm::relax::ConstantNode::TVM_DECLARE_FINAL_OBJECT_INFO ( ConstantNode  ,
LeafExprNode   
)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::relax::ConstantNode::_type_key = "relax.expr.Constant"
staticconstexpr

◆ data

runtime::NDArray tvm::relax::ConstantNode::data

The data of the tensor.


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