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

Call corresponds to callable invocation. Corresponds to operation in computational graph terminology. More...

#include <expr.h>

Inheritance diagram for tvm::relax::CallNode:
Collaboration diagram for tvm::relax::CallNode:

Public Member Functions

 TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.expr.Call", CallNode, ExprNode)
 
- Public Member Functions inherited from tvm::RelaxExprNode
 TVM_FFI_DECLARE_OBJECT_INFO ("ir.RelaxExpr", RelaxExprNode, BaseExprNode)
 
- Public Member Functions inherited from tvm::BaseExprNode
 TVM_FFI_DECLARE_OBJECT_INFO ("ir.BaseExpr", 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

Expr op
 The operator(function) being invoked. More...
 
tvm::ffi::Array< Exprargs
 The arguments(inputs) of the call. More...
 
Attrs attrs
 The additional attributes. More...
 
ffi::Array< StructInfosinfo_args
 The structure info arguments of a CallNode. sinfo_args is designed to be non-empty only for intrinsic op (e.g., call_tir, call_builtin_with_ctx, etc.) and calls to ExternFuncs, with the main usage of structure info inference. More...
 
- Public Attributes inherited from tvm::RelaxExprNode
ffi::Optional< ObjectRef > struct_info_ = ffi::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...
 

Additional Inherited Members

- Static Public Attributes inherited from tvm::RelaxExprNode
static constexpr const uint32_t _type_child_slots = 22
 
- Static Public Attributes inherited from tvm::BaseExprNode
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode
 
static constexpr const uint32_t _type_child_slots = 64
 

Detailed Description

Call corresponds to callable invocation. Corresponds to operation in computational graph terminology.

Member Function Documentation

◆ RegisterReflection()

static void tvm::relax::CallNode::RegisterReflection ( )
inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::relax::CallNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL ( "relax.expr.Call"  ,
CallNode  ,
ExprNode   
)

Member Data Documentation

◆ args

tvm::ffi::Array<Expr> tvm::relax::CallNode::args

The arguments(inputs) of the call.

◆ attrs

Attrs tvm::relax::CallNode::attrs

The additional attributes.

◆ op

Expr tvm::relax::CallNode::op

The operator(function) being invoked.

◆ sinfo_args

ffi::Array<StructInfo> tvm::relax::CallNode::sinfo_args

The structure info arguments of a CallNode. sinfo_args is designed to be non-empty only for intrinsic op (e.g., call_tir, call_builtin_with_ctx, etc.) and calls to ExternFuncs, with the main usage of structure info inference.


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