GetTensor
Description
Obtains the attribute value of type tensor.
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. |
Returns
Pointer to the attribute value
Restrictions
None
Example
const RuntimeAttrs * runtime_attrs = kernel_context->GetAttrs(); const Tensor *attr0 = runtime_attrs->GetTensor(0);
Parent topic: RuntimeAttrs