GetDynamicInputShape

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

const Shape *GetDynamicInputShape(const size_t ir_index, const size_t relative_index) const

Parameter	Input/Output	Description
ir_index	Input	Index of the dynamic input in the operator IR prototype definition, starting from 0.
relative_index	Input	Relative index after the input is instantiated. For example, if three inputs are instantiated for a DYNAMIC_INPUT, the value range of relative_index is [0, 2].

Pointer to a specified input shape. If the input ir_index or relative_index is invalid, a null pointer is returned.

For details about the definition of the shape type, see Shape.

None

ge::graphStatus InferShapeForXXX(InferShapeContext *context) {
  auto in_shape = context->GetDynamicInputShape(2, 2);
  // ...
}

Parent topic: InferShapeContext