GetRequiredInputDataType

Description

Obtains the data type of a required input based on the input index in the operator prototype definition.

Prototype

ge::DataType GetRequiredInputDataType(const size_t ir_index) const

Parameters

Returns

Data type of a specified input. If the input ir_index is invalid, DT_UNDEFINED is returned.

Restrictions

None

Example

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