Managed reference to TensorIntrinNode.
More...
#include <function.h>
Managed reference to TensorIntrinNode.
◆ TensorIntrin()
Constructor.
- Parameters
-
desc | The function to describe the computation. |
impl | The function of the implementation for the execution. |
◆ 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
-
name | The name of the TensorIntrin. |
allow_missing | Whether 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
-
This | method 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
-
name | The name of the TensorIntrin to register |
intrin | The TensorIntrin to register. |
override | Whether override existing intrinsic. |
- Exceptions
-
This | method throws an exception if the TensorIntrin with the specified name already exists. |
◆ TVM_DEFINE_OBJECT_REF_METHODS()
The documentation for this class was generated from the following file: