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) | |
![]() | |
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) | |
![]() | |
virtual | ~OperationNode () |
TVM_FFI_DECLARE_OBJECT_INFO ("te.Operation", OperationNode, Object) | |
Static Public Member Functions | |
static void | RegisterReflection () |
![]() | |
static void | RegisterReflection () |
![]() | |
static void | RegisterReflection () |
Public Attributes | |
ffi::Array< PrimExpr > | body |
the compute expression More... | |
![]() | |
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... | |
![]() | |
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