You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The ONNX ScatterND layer in opset 16 supports a reduction argument.
In onnx2trt, ScatterND is simply lowered to iScatterLayer, which does not take a reduction argument.
We should update this with a clear error message or add reduction support.
Environment
TensorRT Version: 8.6.1 ONNX-TensorRT Version / Branch: 8.6 GA release GPU Type: A4000 Nvidia Driver Version: 535 CUDA Version: 12.2 CUDNN Version: 8.9.5 TensorFlow + TF2ONNX Version (if applicable): 2.15, 1.17
Steps To Reproduce
Use any onnx model with ScatterND max reduction and you can see the TensorRT model does not respect reductions.
The text was updated successfully, but these errors were encountered:
Description
The ONNX ScatterND layer in opset 16 supports a reduction argument.
In onnx2trt, ScatterND is simply lowered to iScatterLayer, which does not take a reduction argument.
We should update this with a clear error message or add reduction support.
Environment
TensorRT Version: 8.6.1
ONNX-TensorRT Version / Branch: 8.6 GA release
GPU Type: A4000
Nvidia Driver Version: 535
CUDA Version: 12.2
CUDNN Version: 8.9.5
TensorFlow + TF2ONNX Version (if applicable): 2.15, 1.17
Steps To Reproduce
Use any onnx model with ScatterND max reduction and you can see the TensorRT model does not respect reductions.
The text was updated successfully, but these errors were encountered: