AddListTensorShape

Description

Adds an element of type TensorShape to the AttrValue attribute in the form of a TensorShape list.

Prototype

std::shared_ptr<TensorShape> AddListTensorShape()

Parameters

None

Returns

An std::shared_ptr<TensorShape> for the pointer to the TensorShape element added to the AttrValue attribute, TensorShape list.

Restrictions

None

Example

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

AttrValue *attr = ctx.GetAttr("attr1");
auto shape = attr->AddListTensorShape();