SetTensor

Description

Sets AttrValue of type tensor.

Prototype

bool SetTensor(const Tensor *tensor)

Parameters

Returns

A bool.

true on success; false on failure.

Restrictions

None

Example

The following gives the prototype definition of an operator whose OpType is A, with attribute name set to attr1 and type Tensor.

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, Tensor, Tensor())
    .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");
bool ret = attr->SetTensor(tensor);