microsoft / microsoft/onnxruntime
Squeeze node fails when axes is ""
- Dominant language
- C++
- Stars
- 21.9k
- Forks
- 4.2k
- Avg merge
- 4d 11h
- Merged PRs (30d)
- 184
Description
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
Contributor guide
Assessment
This issue has not been assessed yet.