GetListTensorShape

Description

Obtains the value of the AttrValue attribute, TensorShape list.

Prototype

std::vector<TensorShape> GetListTensorShape() const

Parameters

None

Returns

std::vector<TensorShape> for the value of the AttrValue attribute, TensorShape list.

Restrictions

None

Example

This following shows how to obtain operator A's attribute value during operator implementation.

AttrValue *attr = ctx.GetAttr("attr1");
std::vector<TensorShape> shapes = attr->GetListTensorShape();