tvm
Public Member Functions | List of all members
tvm::relax::Constant Class Reference

#include <expr.h>

Inheritance diagram for tvm::relax::Constant:
Collaboration diagram for tvm::relax::Constant:

Public Member Functions

 Constant (runtime::NDArray data, Optional< StructInfo > struct_info_annotation=std::nullopt, Span span=Span())
 The constructor. More...
 
 TVM_DEFINE_OBJECT_REF_METHODS (Constant, LeafExpr, ConstantNode)
 
 TVM_DEFINE_OBJECT_REF_COW_METHOD (ConstantNode)
 
- Public Member Functions inherited from tvm::relax::LeafExpr
 TVM_DEFINE_OBJECT_REF_METHODS (LeafExpr, Expr, LeafExprNode)
 
- Public Member Functions inherited from tvm::RelaxExpr
 TVM_DEFINE_OBJECT_REF_METHODS (RelaxExpr, BaseExpr, RelaxExprNode)
 
- Public Member Functions inherited from tvm::BaseExpr
 TVM_DEFINE_OBJECT_REF_METHODS (BaseExpr, ObjectRef, BaseExprNode)
 

Constructor & Destructor Documentation

◆ Constant()

tvm::relax::Constant::Constant ( runtime::NDArray  data,
Optional< StructInfo struct_info_annotation = std::nullopt,
Span  span = Span() 
)
explicit

The constructor.

Parameters
dataThe data of the constant tensor.
struct_info_annotationThe struct info of the constant tensor. If not specified, infer it from data.
spanThe source span of the expression.

Member Function Documentation

◆ TVM_DEFINE_OBJECT_REF_COW_METHOD()

tvm::relax::Constant::TVM_DEFINE_OBJECT_REF_COW_METHOD ( ConstantNode  )

◆ TVM_DEFINE_OBJECT_REF_METHODS()

tvm::relax::Constant::TVM_DEFINE_OBJECT_REF_METHODS ( Constant  ,
LeafExpr  ,
ConstantNode   
)

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