|
tvm
|
Managed reference to ExternOpNode. More...
#include <operation.h>
Public Member Functions | |
| ExternOp (std::string name, std::string tag, ffi::Map< ffi::String, ffi::Any > attrs, ffi::Array< Tensor > inputs, ffi::Array< Buffer > input_placeholders, ffi::Array< Buffer > output_placeholders, Stmt body) | |
| TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (ExternOp, Operation, ExternOpNode) | |
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... | |
Managed reference to ExternOpNode.
| tvm::te::ExternOp::ExternOp | ( | std::string | name, |
| std::string | tag, | ||
| ffi::Map< ffi::String, ffi::Any > | attrs, | ||
| ffi::Array< Tensor > | inputs, | ||
| ffi::Array< Buffer > | input_placeholders, | ||
| ffi::Array< Buffer > | output_placeholders, | ||
| Stmt | body | ||
| ) |
| tvm::te::ExternOp::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE | ( | ExternOp | , |
| Operation | , | ||
| ExternOpNode | |||
| ) |