tvm
Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
tvm::te::ComputeOpNode Class Reference

A Compute op that compute a tensor on certain domain. More...

#include <operation.h>

Inheritance diagram for tvm::te::ComputeOpNode:
Collaboration diagram for tvm::te::ComputeOpNode:

Public Member Functions

 ComputeOpNode ()
 constructor More...
 
int num_outputs () const final
 
DataType output_dtype (size_t i) const final
 Get data type. i-th output tensor. More...
 
ffi::Array< TensorInputTensors () const final
 List all the input Tensors. More...
 
 TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("te.ComputeOp", ComputeOpNode, BaseComputeOpNode)
 
- Public Member Functions inherited from tvm::te::BaseComputeOpNode
ffi::Array< PrimExproutput_shape (size_t idx) const final
 Get shape of i-th output tensor. More...
 
 TVM_FFI_DECLARE_OBJECT_INFO ("te.BaseComputeOp", BaseComputeOpNode, OperationNode)
 
- Public Member Functions inherited from tvm::te::OperationNode
virtual ~OperationNode ()
 
 TVM_FFI_DECLARE_OBJECT_INFO ("te.Operation", OperationNode, Object)
 

Static Public Member Functions

static void RegisterReflection ()
 
- Static Public Member Functions inherited from tvm::te::BaseComputeOpNode
static void RegisterReflection ()
 
- Static Public Member Functions inherited from tvm::te::OperationNode
static void RegisterReflection ()
 

Public Attributes

ffi::Array< PrimExprbody
 the compute expression More...
 
- Public Attributes inherited from tvm::te::BaseComputeOpNode
ffi::Array< IterVaraxis
 IterVar on each axis. More...
 
ffi::Array< IterVarreduce_axis
 IterVar on each reduction axis, if the body is a Reduce. More...
 
- Public Attributes inherited from tvm::te::OperationNode
std::string name
 optional name of the operation More...
 
std::string tag
 optional tag of the operation More...
 
ffi::Map< ffi::String, ffi::Any > attrs
 additional attributes of the operation More...
 

Detailed Description

A Compute op that compute a tensor on certain domain.

Constructor & Destructor Documentation

◆ ComputeOpNode()

tvm::te::ComputeOpNode::ComputeOpNode ( )
inline

constructor

Member Function Documentation

◆ InputTensors()

ffi::Array<Tensor> tvm::te::ComputeOpNode::InputTensors ( ) const
finalvirtual

List all the input Tensors.

Returns
List of input tensors.

Implements tvm::te::OperationNode.

◆ num_outputs()

int tvm::te::ComputeOpNode::num_outputs ( ) const
finalvirtual
Returns
number of outputs

Implements tvm::te::OperationNode.

◆ output_dtype()

DataType tvm::te::ComputeOpNode::output_dtype ( size_t  i) const
finalvirtual

Get data type. i-th output tensor.

Parameters
iThe output index.
Returns
type of i-th output.

Implements tvm::te::OperationNode.

◆ RegisterReflection()

static void tvm::te::ComputeOpNode::RegisterReflection ( )
inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::te::ComputeOpNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL ( "te.ComputeOp"  ,
ComputeOpNode  ,
BaseComputeOpNode   
)

Member Data Documentation

◆ body

ffi::Array<PrimExpr> tvm::te::ComputeOpNode::body

the compute expression


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