tvm
Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
tvm::relax::CallTIRInplaceAttrs Struct Reference

Attributes used in call_tir_inplace. More...

#include <op.h>

Inheritance diagram for tvm::relax::CallTIRInplaceAttrs:
Collaboration diagram for tvm::relax::CallTIRInplaceAttrs:

Public Member Functions

 TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.attrs.CallTIRInplaceAttrs", CallTIRInplaceAttrs, AttrsNode)
 
- Public Member Functions inherited from tvm::AttrsNode
 TVM_FFI_DECLARE_OBJECT_INFO ("ir.Attrs", AttrsNode, ffi::Object)
 

Static Public Member Functions

static void RegisterReflection ()
 

Public Attributes

ffi::Array< int64_t > inplace_indices
 Indices that describe which input corresponds to which output. More...
 

Additional Inherited Members

- Static Public Attributes inherited from tvm::AttrsNode
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode
 

Detailed Description

Attributes used in call_tir_inplace.

Member Function Documentation

◆ RegisterReflection()

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

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::relax::CallTIRInplaceAttrs::TVM_FFI_DECLARE_OBJECT_INFO_FINAL ( "relax.attrs.CallTIRInplaceAttrs"  ,
CallTIRInplaceAttrs  ,
AttrsNode   
)

Member Data Documentation

◆ inplace_indices

ffi::Array<int64_t> tvm::relax::CallTIRInplaceAttrs::inplace_indices

Indices that describe which input corresponds to which output.

If the ith member has the value k >= 0, then that means that input k should be used to store the ith output. If an element has the value -1, that means a new tensor should be allocated for that output.


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