Managed reference to ComputeOpNode. More...
#include <operation.h>
Inheritance diagram for tvm::te::ComputeOp:
Collaboration diagram for tvm::te::ComputeOp:
Public Member Functions | |
| ComputeOp (std::string name, std::string tag, ffi::Map< ffi::String, ffi::Any > attrs, ffi::Array< IterVar > axis, ffi::Array< PrimExpr > body) | |
| TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (ComputeOp, Operation, ComputeOpNode) | |
| TVM_DEFINE_OBJECT_REF_COW_METHOD (ComputeOpNode) | |
 Public Member Functions inherited from tvm::te::Operation | |
| Operation (ffi::UnsafeInit tag) | |
| const OperationNode * | operator-> () const |
| access the internal node container | |
| Tensor | output (size_t i) const |
| get the i-th output of the operation. | |
Additional Inherited Members | |
| Public Types inherited from tvm::te::Operation | |
| using | ContainerType = OperationNode |
| specify container node | |
| Static Public Attributes inherited from tvm::te::Operation | |
| static constexpr bool | _type_is_nullable = false |
Detailed Description
Managed reference to ComputeOpNode.
- See also
- ComputeOpNode
Constructor & Destructor Documentation
◆ ComputeOp()
| tvm::te::ComputeOp::ComputeOp | ( | std::string | name, |
| std::string | tag, | ||
| ffi::Map< ffi::String, ffi::Any > | attrs, | ||
| ffi::Array< IterVar > | axis, | ||
| ffi::Array< PrimExpr > | body | ||
| ) |
Member Function Documentation
◆ TVM_DEFINE_OBJECT_REF_COW_METHOD()
| tvm::te::ComputeOp::TVM_DEFINE_OBJECT_REF_COW_METHOD | ( | ComputeOpNode | ) |
◆ TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE()
| tvm::te::ComputeOp::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE | ( | ComputeOp | , |
| Operation | , | ||
| ComputeOpNode | |||
| ) |
The documentation for this class was generated from the following file:
- include/tvm/te/operation.h
