tvm
Loading...
Searching...
No Matches
Public Member Functions | List of all members
tvm::te::ExternOp Class Reference

Managed reference to ExternOpNode. More...

#include <operation.h>

Inheritance diagram for tvm::te::ExternOp:
Collaboration diagram for tvm::te::ExternOp:

Public Member Functions

 ExternOp (std::string name, std::string tag, ffi::Map< ffi::String, ffi::Any > attrs, ffi::Array< Tensor > inputs, ffi::Array< TensorVar > input_placeholders, ffi::Array< TensorVar > output_placeholders, SeqStmt body)
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (ExternOp, Operation, ExternOpNode)
 
- Public Member Functions inherited from tvm::te::Operation
 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.
 

Additional Inherited Members

- Public Types inherited from tvm::te::Operation
using ContainerType = OperationNode
 specify container node
 
- Static Public Attributes inherited from tvm::te::Operation
static constexpr bool _type_is_nullable = false
 

Detailed Description

Managed reference to ExternOpNode.

See also
ExternOpNode

Constructor & Destructor Documentation

◆ ExternOp()

tvm::te::ExternOp::ExternOp ( std::string  name,
std::string  tag,
ffi::Map< ffi::String, ffi::Any >  attrs,
ffi::Array< Tensor >  inputs,
ffi::Array< TensorVar >  input_placeholders,
ffi::Array< TensorVar >  output_placeholders,
SeqStmt  body 
)

Member Function Documentation

◆ TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE()

tvm::te::ExternOp::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE ( ExternOp  ,
Operation  ,
ExternOpNode   
)

The documentation for this class was generated from the following file: