GetOptionalInputDataType

Function Usage

Obtains the data type of an optional input based on the input index in the operator prototype definition.

Prototype

1

ge::DataType GetOptionalInputDataType(const size_t ir_index) const

Parameters

Returns

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

Constraints

None

Examples

1
2
3
4

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