Bug fix: softmax()'s axis argument should EnforceRange

webmachinelearning / webnn

🧠 Web Neural Network API

https://www.w3.org/TR/webnn/

Other

397 stars 48 forks source link

Bug fix: softmax()'s axis argument should EnforceRange #746

Closed inexorabletash closed 4 months ago

inexorabletash commented 4 months ago

All input integers have [EnforceRange] applied to prevent weirdness... except softmax()'s axis argument. Already present in the Chromium prototype implementation; it looks like this argument was just missed in 6023741.

Preview | Diff

inexorabletash commented 4 months ago

@huningxin and/or @fdwr - can you please review/merge?