tvm::Op Class Reference

Managed reference class to OpNode. More...

#include <op.h>

Inheritance diagram for tvm::Op:

Collaboration diagram for tvm::Op:

Static Public Member Functions
template<typename ValueType >
static OpAttrMap< ValueType >	GetAttrMap (const String &attr_name)
	Get additional registered attribute about operators. If nothing has been registered, an empty OpAttrMap will be returned. More...
static bool	HasAttrMap (const String &attr_name)
	Checks if an attr map is present in the registry. More...
static const Op &	Get (const String &op_name)
	Get an Op for a given operator name. Will raise an error if the op has not been registered. More...

Additional Inherited Members
Public Types inherited from tvm::runtime::ObjectRef
using	ContainerType = Object
	type indicate the container type. More...
Public Member Functions inherited from tvm::RelaxExpr
	TVM_DEFINE_OBJECT_REF_METHODS (RelaxExpr, BaseExpr, RelaxExprNode)
Public Member Functions inherited from tvm::BaseExpr
	TVM_DEFINE_OBJECT_REF_METHODS (BaseExpr, ObjectRef, BaseExprNode)
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 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 class to OpNode.

See also

OpNode

Member Function Documentation

Get()

static const Op& tvm::Op::Get ( const String &  op_name )

static

Get an Op for a given operator name. Will raise an error if the op has not been registered.

Parameters

op_name

Name of the operator.

Returns

Pointer to a Op, valid throughout program lifetime.

GetAttrMap()

template<typename ValueType >

OpAttrMap< ValueType > tvm::Op::GetAttrMap ( const String &  attr_name )

inlinestatic

Get additional registered attribute about operators. If nothing has been registered, an empty OpAttrMap will be returned.

Parameters

attr_name

The name of the attribute.

Returns

An OpAttrMap of specified attr_name.

Template Parameters

ValueType

The type of the attribute.

HasAttrMap()

static bool tvm::Op::HasAttrMap ( const String &  attr_name )

static

Checks if an attr map is present in the registry.

Parameters

attr_name

The name of the attribute.

Returns

bool True if the attr is present.

---

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

• include/tvm/ir/op.h