tvm
|
Managed reference to PlaceholderOpNode. More...
#include <operation.h>
Public Member Functions | |
PlaceholderOp (std::string name, ffi::Array< PrimExpr > shape, DataType dtype) | |
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (PlaceholderOp, Operation, PlaceholderOpNode) | |
![]() | |
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 | |
![]() | |
using | ContainerType = OperationNode |
specify container node More... | |
Managed reference to PlaceholderOpNode.
tvm::te::PlaceholderOp::PlaceholderOp | ( | std::string | name, |
ffi::Array< PrimExpr > | shape, | ||
DataType | dtype | ||
) |
tvm::te::PlaceholderOp::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE | ( | PlaceholderOp | , |
Operation | , | ||
PlaceholderOpNode | |||
) |