CANN元数据算子映射函数

AutoMappingByOpFn【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef函数功能自动映射回调函数。函数原型Status AutoMappingByOpFn(const ge::Operator op_src, ge::Operator op);参数说明参数输入/输出说明op_src输入转换前原始模型中的算子包含原始模型中算子的属性。op输入适配昇腾AI处理器的算子。调用示例原始TensorFlow算子与适配昇腾AI处理器的算子属性一一映射的场景REGISTER_CUSTOM_OP(SoftplusGrad) .FrameworkType(TENSORFLOW) .OriginOpType(SoftplusGrad) .ParseParamsByOperatorFn(AutoMappingByOpFn) .ImplyType(ImplyType::TVM);原始TensorFlow算子与适配昇腾AI处理器的算子属性无法一一映射的场景Status ParseResizeArea(const ge::Operator op_src, ge::Operator op) { AutoMappingByOpFn(op_src, op); ge::TensorDesc input_tensor op.GetInputDesc(images); input_tensor.SetOriginFormat(ge::FORMAT_NHWC); input_tensor.SetFormat(ge::FORMAT_NHWC); auto ret op.UpdateInputDesc(images, input_tensor); if(ret ! ge::GRAPH_SUCCESS){ return FAILED; } ge::TensorDesc output_tensor op.GetOutputDesc(y); output_tensor.SetOriginFormat(ge::FORMAT_NHWC); output_tensor.SetFormat(ge::FORMAT_NHWC); auto ret_output op.UpdateOutputDesc(y, output_tensor); if(ret_output ! ge::GRAPH_SUCCESS){ return FAILED; } return SUCCESS; } // register ResizeArea op to GE REGISTER_CUSTOM_OP(ResizeArea) .FrameworkType(TENSORFLOW) .OriginOpType(ResizeArea) .ParseParamsByOperatorFn(ParseResizeArea) .ImplyType(ImplyType::AI_CPU);【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考