GetTensor
Function Usage
Obtains the attribute value of type tensor.
Prototype
1 | 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
Constraints
None
Examples
1 2 | const RuntimeAttrs * runtime_attrs = kernel_context->GetAttrs(); const Tensor *attr0 = runtime_attrs->GetTensor(0); |
Parent topic: RuntimeAttrs