简介
InferDataTypeContext继承自ExtendedKernelContext，是一个用于数据类型（DataType）推导的上下文类，该类的主要作用是在算子数据类型推导过程中，提供获取输入、输出数据类型的接口，并支持设置输出数据类型。
InferDataTypeContext继承关系图如下：
需要包含的头文件
1 | #include <infer_datatype_context.h> |
Public成员函数
ge::DataType GetInputDataType(const size_t index) const ge::DataType GetOptionalInputDataType(const size_t ir_index) const ge::DataType GetDynamicInputDataType(const size_t ir_index, const size_t relative_index) const ge::DataType GetRequiredInputDataType(const size_t ir_index) const ge::DataType GetOutputDataType(const size_t index) const ge::graphStatus SetOutputDataType(const size_t index, const ge::DataType datatype)
父主题： InferDataTypeContext