tvm
Public Member Functions | Static Public Member Functions | List of all members
tvm::tir::TensorIntrin Class Reference

Managed reference to TensorIntrinNode. More...

#include <function.h>

Inheritance diagram for tvm::tir::TensorIntrin:
Collaboration diagram for tvm::tir::TensorIntrin:

Public Member Functions

 TensorIntrin (PrimFunc desc, PrimFunc impl)
 Constructor. More...
 
 TVM_DEFINE_OBJECT_REF_METHODS (TensorIntrin, ObjectRef, TensorIntrinNode)
 

Static Public Member Functions

static void Register (String name, TensorIntrin intrin, bool override=false)
 Create and register a TensorIntrin. After registration, the TensorIntrin can be looked up with its name. More...
 
static Optional< TensorIntrinGet (String name, bool allow_missing=false)
 Look up TensorIntrin by name. Raises an exception if not found. More...
 

Detailed Description

Managed reference to TensorIntrinNode.

Constructor & Destructor Documentation

◆ TensorIntrin()

tvm::tir::TensorIntrin::TensorIntrin ( PrimFunc  desc,
PrimFunc  impl 
)
explicit

Constructor.

Parameters
descThe function to describe the computation.
implThe function of the implementation for the execution.

Member Function Documentation

◆ Get()

static Optional<TensorIntrin> tvm::tir::TensorIntrin::Get ( String  name,
bool  allow_missing = false 
)
static

Look up TensorIntrin by name. Raises an exception if not found.

Parameters
nameThe name of the TensorIntrin.
allow_missingWhether to allow missing tensor intrin. If false, an exception is raised if the tensor intrin is not found.
Returns
The TensorIntrin with the specified name.
Exceptions
Thismethod throws an exception if the TensorIntrin does not exist and allow_missing is false.

◆ Register()

static void tvm::tir::TensorIntrin::Register ( String  name,
TensorIntrin  intrin,
bool  override = false 
)
static

Create and register a TensorIntrin. After registration, the TensorIntrin can be looked up with its name.

Parameters
nameThe name of the TensorIntrin to register
intrinThe TensorIntrin to register.
overrideWhether override existing intrinsic.
Exceptions
Thismethod throws an exception if the TensorIntrin with the specified name already exists.

◆ TVM_DEFINE_OBJECT_REF_METHODS()

tvm::tir::TensorIntrin::TVM_DEFINE_OBJECT_REF_METHODS ( TensorIntrin  ,
ObjectRef  ,
TensorIntrinNode   
)

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