Managed reference to ScanOpNode. More...
#include <operation.h>
Inheritance diagram for tvm::te::ScanOp:
Collaboration diagram for tvm::te::ScanOp:
Public Member Functions | |
| ScanOp (std::string name, std::string tag, ffi::Optional< ffi::Map< ffi::String, ffi::Any >> attrs, IterVar axis, ffi::Array< Tensor > init, ffi::Array< Tensor > update, ffi::Array< Tensor > state_placeholder, ffi::Array< Tensor > input) | |
| TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (ScanOp, Operation, ScanOpNode) | |
 Public Member Functions inherited from tvm::te::Operation | |
| Operation () | |
| default constructor More... | |
| Operation (ObjectPtr< Object > n) | |
| Operation (ffi::UnsafeInit tag) | |
| const OperationNode * | operator-> () const |
| access the internal node container More... | |
| Tensor | output (size_t i) const |
| get the i-th output of the operation. More... | |
Additional Inherited Members | |
| Public Types inherited from tvm::te::Operation | |
| using | ContainerType = OperationNode |
| specify container node More... | |
Detailed Description
Managed reference to ScanOpNode.
- See also
- ScanOpNode
Constructor & Destructor Documentation
◆ ScanOp()
| tvm::te::ScanOp::ScanOp | ( | std::string | name, |
| std::string | tag, | ||
| ffi::Optional< ffi::Map< ffi::String, ffi::Any >> | attrs, | ||
| IterVar | axis, | ||
| ffi::Array< Tensor > | init, | ||
| ffi::Array< Tensor > | update, | ||
| ffi::Array< Tensor > | state_placeholder, | ||
| ffi::Array< Tensor > | input | ||
| ) |
Member Function Documentation
◆ TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE()
| tvm::te::ScanOp::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE | ( | ScanOp | , |
| Operation | , | ||
| ScanOpNode | |||
| ) |
The documentation for this class was generated from the following file:
- include/tvm/te/operation.h
