From 928d3a18b18d826614fb898c34577e89fbd5d54e Mon Sep 17 00:00:00 2001
From: Joshua Bell <jsbell@chromium.org>
Date: Tue, 30 Jul 2024 17:59:21 -0700
Subject: [PATCH] Bug fix: softmax()'s axis argument should EnforceRange (#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.
---
 index.bs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/index.bs b/index.bs
index 2e102d21..caa9a5dd 100644
--- a/index.bs
+++ b/index.bs
@@ -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>