Error in PredictCost() for the op: op: "Softmax" attr

[krishnanNuance]

Describe the bug

The model converted from ONNX to TF on validation returns NaN as result.

To Reproduce

1. export fairseq's transformer lm model to onnx
2. export the onnx model to TF
3. load the saved_model and validate with any src_token value
4. the output array contains NaN

Python, ONNX, ONNX-TF, Tensorflow version

• Python version: 1.11.0+cu102
• ONNX version: 1.10.2
• ONNX-TF version: 1.10.0 + implementation for squeeze and unsqueeze for opset v 13
• Tensorflow version: 2.8.0