add WebNN in log

shiyi9801 · Oct 24, 2024 · 1ccd360 · 1ccd360
1 parent 9f62135
commit 1ccd360
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/js/web/docs/webnn-operators.md b/js/web/docs/webnn-operators.md
@@ -80,6 +80,7 @@ operators and the supported opset domain/versions in **WebNN EP** by ONNX Runtim
 | Relu | ai.onnx(7-12, 13, 14+) | relu | ✓ | ✓ | |
 | Reshape | ai.onnx(7-12, 13, 14-18, 19-20, 21+) | reshape | ✓ | ✓ | Input 'shape' should be a constant, 0 dimension value in 'shape' is not supported |
 | Resize | ai.onnx(11-12, 13-17, 18, 19+) | resample2d | ✓ | ✓ | Only supports 4-D input, antialias == 0, coordinate_transformation_mode == 'half_pixel', exclude_outside == 0, keep_aspect_ratio_policy == 'stretch', 'linear' and 'nearest' modes, input 'scales' and 'sizes' if present must be a constant |
+| ScatterND | ai.onnx(11-12, 13-15, 16-17, 18+) | scatterND | ✗ | ✓ | Only supports 'reduction' == 'none' |
 | Shape | ai.onnx(7-12, 13-14, 15-18, 19-20, 21+) | slice | ✓ | ✓ | |
 | Sigmoid | ai.onnx(7-12, 13+) | sigmoid | ✓ | ✓ | |
 | Softplus | ai.onnx(7+) | softplus | ✓ | ✓ | |

diff --git a/onnxruntime/core/providers/webnn/builders/impl/gatherND_op_builder.cc b/onnxruntime/core/providers/webnn/builders/impl/gatherND_op_builder.cc
@@ -48,7 +48,7 @@ bool GatherNDOpBuilder::IsOpSupportedImpl(const InitializedTensorSet& /* initial
                                           const logging::Logger& logger) const {
   NodeAttrHelper helper(node);
   if (helper.Get("batch_dims", 0) != 0) {
-    LOGS(logger, VERBOSE) << "GatherND: batch_dims must be equal to 0";
+    LOGS(logger, VERBOSE) << "GatherND: WebNN only supports batch_dims 0 (default)";
     return false;
   }
 

diff --git a/onnxruntime/core/providers/webnn/builders/impl/scatterND_op_builder.cc b/onnxruntime/core/providers/webnn/builders/impl/scatterND_op_builder.cc
@@ -50,7 +50,7 @@ bool ScatterNDOpBuilder::IsOpSupportedImpl(const InitializedTensorSet& /* initia
                                            const logging::Logger& logger) const {
   NodeAttrHelper helper(node);
   if (helper.Get("reduction", "none") != "none") {
-    LOGS(logger, VERBOSE) << "ScatterND: the type of reduction must be none (default)";
+    LOGS(logger, VERBOSE) << "ScatterND: WebNN only supports reduction type none (default)";
     return false;
   }