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_DEFINE_OBJECT_REF_METHODS (Op, RelaxExpr, OpNode)
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)

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

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.

TVM_DEFINE_OBJECT_REF_METHODS()

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

---

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

• include/tvm/ir/op.h