Skip to content

Commit

Permalink
Bug fix: softmax()'s axis argument should EnforceRange (webmachinelea…
Browse files Browse the repository at this point in the history
…rning#746)

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.
  • Loading branch information
inexorabletash authored Jul 31, 2024
1 parent 4697a0d commit 928d3a1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion index.bs
Original file line number Diff line number Diff line change
Expand Up @@ -5441,7 +5441,7 @@ the N-D input tensor along the given axis.
<script type=idl>
partial interface MLGraphBuilder {
MLOperand softmax(MLOperand input,
unsigned long axis,
[EnforceRange] unsigned long axis,
optional MLOperatorOptions options = {});
};
</script>
Expand Down

0 comments on commit 928d3a1

Please sign in to comment.