|
tvm
|
External computation that cannot be splitted. More...
#include <operation.h>
Public Member Functions | |
| ExternOpNode () | |
| 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< PrimExpr > | output_shape (size_t i) const final |
| Get shape of i-th output tensor. More... | |
| ffi::Array< Tensor > | InputTensors () const final |
| List all the input Tensors. More... | |
| TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("te.ExternOp", ExternOpNode, 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::OperationNode | |
| static void | RegisterReflection () |
Public Attributes | |
| ffi::Array< Tensor > | inputs |
| The input tensors. More... | |
| ffi::Array< Buffer > | input_placeholders |
| Symbolic placeholder representation of inputs. More... | |
| ffi::Array< Buffer > | output_placeholders |
| Symbolic placeholder representation of outputs. More... | |
| Stmt | body |
| the statement that generates the computation. 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... | |
External computation that cannot be splitted.
|
inline |
constructor
|
finalvirtual |
|
finalvirtual |
Implements tvm::te::OperationNode.
|
finalvirtual |
Get data type. i-th output tensor.
| i | The output index. |
Implements tvm::te::OperationNode.
|
finalvirtual |
Get shape of i-th output tensor.
| i | The output index. |
Implements tvm::te::OperationNode.
|
inlinestatic |
| tvm::te::ExternOpNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL | ( | "te.ExternOp" | , |
| ExternOpNode | , | ||
| OperationNode | |||
| ) |
| Stmt tvm::te::ExternOpNode::body |
the statement that generates the computation.
| ffi::Array<Buffer> tvm::te::ExternOpNode::input_placeholders |
Symbolic placeholder representation of inputs.
| ffi::Array<Tensor> tvm::te::ExternOpNode::inputs |
The input tensors.
| ffi::Array<Buffer> tvm::te::ExternOpNode::output_placeholders |
Symbolic placeholder representation of outputs.