GetRequiredOutputFormat

Function Usage

Obtains the pointer to a required output format based on the output index in the operator prototype definition.

Prototype

1
StorageFormat *GetRequiredOutputFormat(const size_t ir_index)

Parameters

Parameter

Input/Output

Description

ir_index

Input

Index in the IR prototype definition, starting from 0.

Returns

Output format pointer. If the ir_index is invalid, a null pointer is returned.

For details about the definition of the StorageFormat type, see StorageFormat.

Constraints

None

Examples

1
2
3
4
ge::graphStatus InferFormatForXXX(InferFormatContext *context) {
  auto format = context->GetRequiredOutputFormat(0);
   // ...
}