Skip to content

Squeeze node fails when axes is "" #21661

Open
@justinchuby

Description

@justinchuby

Error [ONNXRuntimeError] : 6 : RUNTIME_EXCEPTION : Non-zero status code returned while running Squeeze node. Name:'node_Squeeze_141' Status Message: /onnxruntime_src/onnxruntime/core/providers/cuda/tensor/squeeze.cc:50 virtual onnxruntime::common::Status onnxruntime::cuda::Squeeze::ComputeInternal(onnxruntime::OpKernelContext*) const axes_tensor != nullptr was false. Axes input is null_ is raise when Squeeze node has a single input. According to the spec https://onnx.ai/onnx/operators/onnx__Squeeze.html, If axes is not provided, all the single dimensions will be removed from the shape. If an axis is selected with shape entry not equal to one, an error is raised.

cc @xadupre @gramalingam

Metadata

Metadata

Assignees

Labels

converter:dynamoissues related supporting the PyTorch Dynamo exporter

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions