onnx / onnx-tensorrt

ONNX-TensorRT: TensorRT backend for ONNX
Apache License 2.0
2.92k stars 542 forks source link

Reduction not supported with ScatterNd in opset 16 #957

Open aboubezari opened 6 months ago

aboubezari commented 6 months ago

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.