tvm
|
Managed reference to ExternOpNode. More...
#include <operation.h>
Public Member Functions | |
ExternOp (std::string name, std::string tag, Map< String, ffi::Any > attrs, Array< Tensor > inputs, Array< Buffer > input_placeholders, Array< Buffer > output_placeholders, Stmt body) | |
TVM_DEFINE_OBJECT_REF_METHODS (ExternOp, Operation, ExternOpNode) | |
![]() | |
Operation () | |
default constructor More... | |
Operation (ObjectPtr< Object > n) | |
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 | |
![]() | |
using | ContainerType = OperationNode |
specify container node More... | |
Managed reference to ExternOpNode.
tvm::te::ExternOp::ExternOp | ( | std::string | name, |
std::string | tag, | ||
Map< String, ffi::Any > | attrs, | ||
Array< Tensor > | inputs, | ||
Array< Buffer > | input_placeholders, | ||
Array< Buffer > | output_placeholders, | ||
Stmt | body | ||
) |
tvm::te::ExternOp::TVM_DEFINE_OBJECT_REF_METHODS | ( | ExternOp | , |
Operation | , | ||
ExternOpNode | |||
) |