static ge::graphStatus InferDataTypeThreshold(gert::InferDataTypeContext* context)
{
OP_LOGD(context->GetNodeName(), "Begin to do InferDataTypeThreshold");
context->SetOutputDataType(IDX_0, context->GetInputDataType(IDX_0));
OP_LOGD(context->GetNodeName(), "End to do InferDataTypeThreshold");
return ge::GRAPH_SUCCESS;
}
Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
Describe the current behavior / 问题描述 (Mandatory / 必填)
Threshold缺少 InferDataType 注册
Environment / 环境信息 (Mandatory / 必填)
950
Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)
验证950的geir通路
Describe the expected behavior / 预期结果 (Mandatory / 必填)
补充输出 dtype 推导（输出 dtype = 输入 dtype）并挂接到IMPL_OP注册链。
Related log / screenshot / 日志 / 截图 (Mandatory / 必填)
static ge::graphStatus InferDataTypeThreshold(gert::InferDataTypeContext* context)
{
OP_LOGD(context->GetNodeName(), "Begin to do InferDataTypeThreshold");
context->SetOutputDataType(IDX_0, context->GetInputDataType(IDX_0));
OP_LOGD(context->GetNodeName(), "End to do InferDataTypeThreshold");
return ge::GRAPH_SUCCESS;
}
IMPL_OP(Threshold).InferDataType(InferDataTypeThreshold);
Special notes for this issue/备注 (Optional / 选填)