Skip to content

When the axes is empty, "set_value_with_tensor_" will report a segmentation fault. #1613

@NewZxy

Description

@NewZxy

Please fill in the information below so that we can solve the problem quickly, Thanks !

Describe the bug
A clear and concise description of what the bug is.
ppdoclayoutv2 模型的json文件中存在算子 set_value_with_tensor_ 的属性 axes是空,导出onnx时:
std::string axes = helper_->Constant(
{1}, ONNX_NAMESPACE::TensorProto::INT64, int64_t(axes_[0]));

这一行代码会导致 segmentation fault.
因为文档中没有找到对应算子介绍,不清楚具体功能,当为空时,当前默认给的0轴,最终导出的onnx加载时会报一些其他算子的错误,怀疑与此有关

请问,这个算子主要的功能是什么,这里是否是bug,如果是,为空时应该怎么处理,谢谢!!!
Informations (please complete the following information):

  • Inference engine for deployment:
    paddle2onnx: 2.0.1
    CPU:arm
  • Why convert to onnx:
    ppdoclayoutv2 不支持npu,先转成 onnx,再转成om
  • Paddle2ONNX Version:
  • Email/Wechat/Phone:

Screenshots
无法上传图片

Additional context

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions