目录

GetTensor

Function Description

Obtains the attribute value of type tensor.

Function Prototype

const Tensor *GetTensor(const size_t index) const

Parameters

Parameter Input/Output Description
index Input Index of the attribute in the IR prototype definition and OP_IMPL registration.

Return Value

Pointer to the attribute value.

Restrictions

None

Calling Example

const RuntimeAttrs * runtime_attrs = kernel_context->GetAttrs();
const Tensor *attr0 = runtime_attrs->GetTensor(0);