GetRequiredInputShape

Description

Obtains the pointer to a required input shape based on the input index in the operator prototype definition.

Prototype

const StorageShape *GetRequiredInputShape(const size_t ir_index) const

Parameters

Returns

Pointer to the specified input shape. The input shape contains the original shape and storage shape information. For details about the definition of the StorageShape type, see StorageShape.

If the input ir_index is invalid, a null pointer is returned.

Restrictions

None

Example

ge::graphStatus InferShape4ConcatD(TilingContext* context) {
  auto in_shape = context->GetRequiredInputShape(0);
  ...
}