Managed reference to ScanOpNode.
More...
#include <operation.h>
|
| | 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) |
| |
| | Operation () |
| | default constructor
|
| |
| | Operation (ffi::ObjectPtr< ffi::Object > n) |
| |
| | 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.
|
| |
Managed reference to ScanOpNode.
- See also
- ScanOpNode
◆ 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 |
|
) |
| |
◆ TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE()
The documentation for this class was generated from the following file: