GetInputDataType

Description

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

Prototype

ge::DataType GetInputDataType(const size_t index) const

Parameters

Returns

Data type of the specified input

If the input index is invalid, DT_UNDEFINED is returned.

Restrictions

None

Example

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