AddListInt
Description
Adds an int to the list of ints for the attribute value.
Prototype
void AddListInt(int64_t i)
Parameters
|
Parameter |
Input/Output |
Description |
|---|---|---|
|
i |
Input |
Int |
Returns
None
Restrictions
None
Example
The following gives the prototype definition of an operator whose OpType is A, with attribute name set to attr1 and type ListInt.
For details about the supported data types of the attribute in the .ATTR API, see API Description.
REG_OP(A)
.INPUT(x, TensorType({ DT_FLOAT }))
.OUTPUT(y, TensorType({ DT_FLOAT }))
.ATTR(attr1, ListInt, {})
.OP_END_FACTORY_REG(A)
In this case, you can use the following API during operator implementation to set operator A's attribute value.
AttrValue *attr = ctx.GetAttr("attr1");
attr->AddListInt(1);
Parent topic: Class AttrValue