PromoteType

功能说明

不同数据类型参数进行运算时，推导应该将类型提升到何种数据类型进行计算。例如float16与float32进行运算，应该将float16先提升至float32后再做运算。

函数原型

DataType PromoteType(const DataType type_a, const DataType type_b)

参数说明

返回值说明

若无法推导类型提升后的数据类型，运算会失败并返回DT_UNDEFINED。

约束说明

无

调用示例

1
2
3
4
5
6

// 判断dtype是否可以与Float32计算，不能则返回
void Func(const DataType dtype) {
    if (PromoteType(dtype, DT_FLOAT) == DT_UNDEFINED) {
        return;
    }
}