GetDynamicInputDataType
Description
Obtains the data type of the dynamic input based on the input index in the operator prototype definition.
Prototype
ge::DataType GetDynamicInputDataType(const size_t ir_index, const size_t relative_index) const
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
ir_index |
Input |
Index of the dynamic input in the operator IR prototype definition, starting from 0. |
relative_index |
Input |
Relative index after the input is instantiated. For example, if three inputs are instantiated for a DYNAMIC_INPUT, the value range of relative_index is [0, 2]. |
Returns
Data type of a specified input. If the input ir_index or relative_index is invalid, DT_UNDEFINED is returned.
Restrictions
None
Example
ge::graphStatus InferDataTypeForXXX(InferDataTypeContext *context) {
auto data_type = context->GetDynamicInputDataType(1, 2);
...
}
Parent topic: InferDataTypeContext