#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...

Public Member Functions inherited from tvm::runtime::ObjectRef
	ObjectRef ()=default
	default constructor More...

	ObjectRef (ObjectPtr< Object > data)
	Constructor from existing object ptr. More...

bool	same_as (const ObjectRef &other) const
	Comparator. More...

bool	operator== (const ObjectRef &other) const
	Comparator. More...

bool	operator!= (const ObjectRef &other) const
	Comparator. More...

bool	operator< (const ObjectRef &other) const
	Comparator. More...

bool	defined () const

const Object *	get () const

const Object *	operator-> () const

bool	unique () const

int	use_count () const

template<typename ObjectType , typename = std::enable_if_t<std::is_base_of_v<Object, ObjectType>>>
const ObjectType *	as () const
	Try to downcast the internal Object to a raw pointer of a corresponding type. More...

template<typename ObjectRefType , typename = std::enable_if_t<std::is_base_of_v<ObjectRef, ObjectRefType>>>
Optional< ObjectRefType >	as () const
	Try to downcast the ObjectRef to a Optional<T> of the requested type. More...

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< TensorIntrin >	Get (String name, bool allow_missing=false)
	Look up TensorIntrin by name. Raises an exception if not found. More...

Additional Inherited Members
Public Types inherited from tvm::runtime::ObjectRef
using	ContainerType = Object
	type indicate the container type. More...

Static Public Attributes inherited from tvm::runtime::ObjectRef
static constexpr bool	_type_is_nullable = true

Protected Member Functions inherited from tvm::runtime::ObjectRef
Object *	get_mutable () const

Static Protected Member Functions inherited from tvm::runtime::ObjectRef
template<typename T >
static T	DowncastNoCheck (ObjectRef ref)
	Internal helper function downcast a ref without check. More...

static void	FFIClearAfterMove (ObjectRef *ref)
	Clear the object ref data field without DecRef after we successfully moved the field. More...

template<typename ObjectType >
static ObjectPtr< ObjectType >	GetDataPtr (const ObjectRef &ref)
	Internal helper function get data_ as ObjectPtr of ObjectType. More...

Protected Attributes inherited from tvm::runtime::ObjectRef
ObjectPtr< Object >	data_
	Internal pointer that backs the reference. More...

Detailed Description

Managed reference to TensorIntrinNode.

Constructor & Destructor Documentation

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

explicit

Parameters

desc	The function to describe the computation.
impl	The function of the implementation for the execution.

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.

Exceptions

This	method throws an exception if the TensorIntrin does not exist and allow_missing is false.

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.

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