|
tvm
|
A Compute op that compute a tensor on certain domain. More...
#include <operation.h>
Public Member Functions | |
| ComputeOpNode () | |
| constructor More... | |
| int | num_outputs () const final |
| DataType | output_dtype (size_t i) const final |
| Get data type. i-th output tensor. More... | |
| ffi::Array< Tensor > | InputTensors () const final |
| List all the input Tensors. More... | |
| TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("te.ComputeOp", ComputeOpNode, BaseComputeOpNode) | |
Public Member Functions inherited from tvm::te::BaseComputeOpNode | |
| ffi::Array< PrimExpr > | output_shape (size_t idx) const final |
| Get shape of i-th output tensor. More... | |
| TVM_FFI_DECLARE_OBJECT_INFO ("te.BaseComputeOp", BaseComputeOpNode, OperationNode) | |
Public Member Functions inherited from tvm::te::OperationNode | |
| virtual | ~OperationNode () |
| TVM_FFI_DECLARE_OBJECT_INFO ("te.Operation", OperationNode, Object) | |
Static Public Member Functions | |
| static void | RegisterReflection () |
Static Public Member Functions inherited from tvm::te::BaseComputeOpNode | |
| static void | RegisterReflection () |
Static Public Member Functions inherited from tvm::te::OperationNode | |
| static void | RegisterReflection () |
Public Attributes | |
| ffi::Array< PrimExpr > | body |
| the compute expression More... | |
Public Attributes inherited from tvm::te::BaseComputeOpNode | |
| ffi::Array< IterVar > | axis |
| IterVar on each axis. More... | |
| ffi::Array< IterVar > | reduce_axis |
| IterVar on each reduction axis, if the body is a Reduce. More... | |
Public Attributes inherited from tvm::te::OperationNode | |
| std::string | name |
| optional name of the operation More... | |
| std::string | tag |
| optional tag of the operation More... | |
| ffi::Map< ffi::String, ffi::Any > | attrs |
| additional attributes of the operation More... | |
A Compute op that compute a tensor on certain domain.
|
inline |
constructor
|
finalvirtual |
|
finalvirtual |
Implements tvm::te::OperationNode.
|
finalvirtual |
Get data type. i-th output tensor.
| i | The output index. |
Implements tvm::te::OperationNode.
|
inlinestatic |
| tvm::te::ComputeOpNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL | ( | "te.ComputeOp" | , |
| ComputeOpNode | , | ||
| BaseComputeOpNode | |||
| ) |
| ffi::Array<PrimExpr> tvm::te::ComputeOpNode::body |
the compute expression