The tensor argument information. More...
#include <arg_info.h>
Inheritance diagram for tvm::meta_schedule::TensorInfoNode:
Collaboration diagram for tvm::meta_schedule::TensorInfoNode:
Public Member Functions | |
| TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("meta_schedule.TensorInfo", TensorInfoNode, ArgInfoNode) | |
| ObjectRef | AsJSON () const |
| Converts the ArgInfo to its corresponding JSON representation. More... | |
 Public Member Functions inherited from tvm::meta_schedule::ArgInfoNode | |
| TVM_FFI_DECLARE_OBJECT_INFO ("meta_schedule.ArgInfo", ArgInfoNode, runtime::Object) | |
| virtual | ~ArgInfoNode ()=default |
| Default destructor. More... | |
Static Public Member Functions | |
| static void | RegisterReflection () |
Public Attributes | |
| runtime::DataType | dtype |
| The data type of the tensor. More... | |
| ffi::Shape | shape |
| The shape of the tensor. More... | |
Detailed Description
The tensor argument information.
Member Function Documentation
◆ AsJSON()
|
virtual |
Converts the ArgInfo to its corresponding JSON representation.
Implements tvm::meta_schedule::ArgInfoNode.
◆ RegisterReflection()
|
inlinestatic |
◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()
| tvm::meta_schedule::TensorInfoNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL | ( | "meta_schedule.TensorInfo" | , |
| TensorInfoNode | , | ||
| ArgInfoNode | |||
| ) |
Member Data Documentation
◆ dtype
| runtime::DataType tvm::meta_schedule::TensorInfoNode::dtype |
The data type of the tensor.
◆ shape
| ffi::Shape tvm::meta_schedule::TensorInfoNode::shape |
The shape of the tensor.
The documentation for this class was generated from the following file:
- include/tvm/meta_schedule/arg_info.h
