Function Description
Obtains the data type of a required input based on the input index in the operator prototype definition.
Function Prototype
ge::DataType GetRequiredInputDataType(const size_t ir_index) const;
Parameters
| Parameter | Input/Output | Description |
|---|
| ir_index | Input | Index of the required input in the operator IR prototype definition, starting from 0. |
Return Value
Data type of a specified input. If the input ir_index is invalid, DT_UNDEFINED is returned.
Calling Example
ge::graphStatus InferDataTypeForXXX(InferDataTypeContext *context) {
auto data_type = context->GetRequiredInputDataType(1);
// ...
}