search

microsoft / onnxruntime-extensions

onnxruntime-extensions: A specialized pre- and post- processing library for ONNX Runtime
MIT License
295 stars 80 forks source link

Add custom kernel ScatterNDOfShape #705

Closed xadupre closed 3 weeks ago

xadupre commented 2 months ago

It combines ScatterND(ConstantOfShape(value=0), indices, udpates, reduction=add).