A Compute op that compute a tensor on certain domain. This is the base class for ComputeOp (operating on a scalar at a time) More...
#include <operation.h>
Inheritance diagram for tvm::te::BaseComputeOpNode:
Collaboration diagram for tvm::te::BaseComputeOpNode:
Public Member Functions | |
| Array< PrimExpr > | output_shape (size_t idx) const final |
| Get shape of i-th output tensor. More... | |
| TVM_DECLARE_BASE_OBJECT_INFO (BaseComputeOpNode, OperationNode) | |
 Public Member Functions inherited from tvm::te::OperationNode | |
| virtual | ~OperationNode () |
| virtual int | num_outputs () const =0 |
| virtual DataType | output_dtype (size_t i) const =0 |
| Get data type. i-th output tensor. More... | |
| virtual Array< Tensor > | InputTensors () const =0 |
| List all the input Tensors. More... | |
| TVM_DECLARE_BASE_OBJECT_INFO (OperationNode, Object) | |
Static Public Member Functions | |
| static void | RegisterReflection () |
| Static Public Member Functions inherited from tvm::te::OperationNode | |
| static void | RegisterReflection () |
Public Attributes | |
| Array< IterVar > | axis |
| IterVar on each axis. More... | |
| 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... | |
| Map< String, ffi::Any > | attrs |
| additional attributes of the operation More... | |
Static Public Attributes | |
| static constexpr const char * | _type_key = "te.BaseComputeOp" |
| Static Public Attributes inherited from tvm::te::OperationNode | |
| static constexpr const char * | _type_key = "te.Operation" |
Detailed Description
A Compute op that compute a tensor on certain domain. This is the base class for ComputeOp (operating on a scalar at a time)
Member Function Documentation
◆ output_shape()
|
finalvirtual |
Get shape of i-th output tensor.
- Parameters
-
i The output index.
- Returns
- shape of i-th output.
Implements tvm::te::OperationNode.
◆ RegisterReflection()
|
inlinestatic |
◆ TVM_DECLARE_BASE_OBJECT_INFO()
| tvm::te::BaseComputeOpNode::TVM_DECLARE_BASE_OBJECT_INFO | ( | BaseComputeOpNode | , |
| OperationNode | |||
| ) |
Member Data Documentation
◆ _type_key
|
staticconstexpr |
◆ axis
| Array<IterVar> tvm::te::BaseComputeOpNode::axis |
IterVar on each axis.
◆ reduce_axis
| Array<IterVar> tvm::te::BaseComputeOpNode::reduce_axis |
IterVar on each reduction axis, if the body is a Reduce.
The documentation for this class was generated from the following file:
- include/tvm/te/operation.h
