tvm
Public Member Functions | List of all members
tvm::te::ComputeOp Class Reference

Managed reference to ComputeOpNode. More...

#include <operation.h>

Inheritance diagram for tvm::te::ComputeOp:
Collaboration diagram for tvm::te::ComputeOp:

Public Member Functions

 ComputeOp (std::string name, std::string tag, ffi::Map< ffi::String, ffi::Any > attrs, ffi::Array< IterVar > axis, ffi::Array< PrimExpr > body)
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (ComputeOp, Operation, ComputeOpNode)
 
 TVM_DEFINE_OBJECT_REF_COW_METHOD (ComputeOpNode)
 
- Public Member Functions inherited from tvm::te::Operation
 Operation ()
 default constructor
More...
 
 Operation (ObjectPtr< Object > n)
 
 Operation (ffi::UnsafeInit tag)
 
const OperationNodeoperator-> () 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...
 

Detailed Description

Managed reference to ComputeOpNode.

See also
ComputeOpNode

Constructor & Destructor Documentation

◆ ComputeOp()

tvm::te::ComputeOp::ComputeOp ( std::string  name,
std::string  tag,
ffi::Map< ffi::String, ffi::Any >  attrs,
ffi::Array< IterVar axis,
ffi::Array< PrimExpr body 
)

Member Function Documentation

◆ TVM_DEFINE_OBJECT_REF_COW_METHOD()

tvm::te::ComputeOp::TVM_DEFINE_OBJECT_REF_COW_METHOD ( ComputeOpNode  )

◆ TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE()

tvm::te::ComputeOp::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE ( ComputeOp  ,
Operation  ,
ComputeOpNode   
)

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