函数原型
ge::graphStatus SetOutputDataType(const size_t index, const ge::DataType datatype);
参数说明
| 参数 | 输入/输出 | 说明 |
|---|
| index | 输入 | 算子IR原型定义中的输出索引，从0开始计数。 |
| datatype | 输入 | 需要设置的输出数据类型。 关于DataType的说明，请参见DataType。 |
调用示例
ge::graphStatus InferDataTypeForXXX(InferDataTypeContext *context) {
auto ret = context->SetOutputDataType(0, ge::DataType::DT_FLOAT);
// ...
}