ATC工具 pb模型转om失败:strided_slice not support, end shape begin shape length mismatch
收藏回复举报
ATC工具 pb模型转om失败:strided_slice not support, end shape begin shape length mismatch
t('forum.solved') 已解决
新人帖
发表于2023-02-06 17:45:14
0 查看

转换日志和模型如下:

链接: https://pan.baidu.com/s/1c4atLI-ZVGDHkMxJg9mLiQ?pwd=4a9a

提取码: 4a9a

pb模型测试正确,可以正常工作 转换命令如下

atc --input_shape="x:1,512,640,3" --input_format=NHWC --output=./Weather_pb --soc_version=Ascend310 --insert_op_conf=./insert_op.cfg --framework=3 --save_original_model=false --model=./Twostage.pb --log=debug

报错信息如下

E89999: Inner Error!
        op[PartitionedCall/model/tf.strided_slice/StridedSlice], not support this type[FUNC:GetConstValue][FILE:util.cc][LINE:830]
        op[PartitionedCall/model/tf.strided_slice/StridedSlice], end shape, begin shape length mismatch![FUNC:StridedSliceInferShape][FILE:selection_ops.cc][LINE:1621]
        PartitionedCall/model/tf.strided_slice/StridedSlice(StridedSlice) call infer function failed.[FUNC:InferShapeAndType][FILE:shape_refiner.cc][LINE:835]
        process pass InferShapePass on node:PartitionedCall/model/tf.strided_slice/StridedSlice failed, ret:1343242270[FUNC:RunPasses][FILE:base_pass.cc][LINE:161]
        build graph failed, graph id:0, ret:1343242268[FUNC:BuildModel][FILE:ge_generator.cc][LINE:1157]

我要发帖子