GetOutputDataType

Description

Obtains the output data type based on the operator output index. The output index refers to the actual index after operator instantiation, not the index in the prototype definition.

Prototype

ge::DataType GetOutputDataType(const size_t index) const

Parameters

Returns

Data type of a specified output. If index is invalid, DT_UNDEFINED is returned.

Restrictions

None

Example

ge::graphStatus InferDataTypeForXXX(InferDataTypeContext *context) {
  auto data_type = context->GetOutputDataType(0);
  ...
}