tvm::Op Class Reference

Managed reference class to OpNode. More...

#include <op.h>

Inheritance diagram for tvm::Op:

Collaboration diagram for tvm::Op:

Public Member Functions
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (Op, RelaxExpr, OpNode)
Public Member Functions inherited from tvm::RelaxExpr
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (RelaxExpr, BaseExpr, RelaxExprNode)
Public Member Functions inherited from tvm::BaseExpr
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (BaseExpr, ObjectRef, BaseExprNode)

Static Public Member Functions
template<typename ValueType >
static OpAttrMap< ValueType >	GetAttrMap (const ffi::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 ffi::String &attr_name)
	Checks if an attr map is present in the registry. More...
static const Op &	Get (const ffi::String &op_name)
	Get an Op for a given operator name. Will raise an error if the op has not been registered. More...

Detailed Description

Managed reference class to OpNode.

See also

OpNode

Member Function Documentation

Get()

static const Op& tvm::Op::Get ( const ffi::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 ffi::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 ffi::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.

TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE()

tvm::Op::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE	(	Op	,
		RelaxExpr	,
		OpNode
	)

---

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

• include/tvm/ir/op.h