webnn: Enforce input data type constraints for conv2d

As specified in webmachinelearning/webnn#646 Bug: 328567884 Change-Id: I620653f574303e6f5b18d3b540b647f07e0cf64c
web-platform-tests · Apr 26, 2024 · 79baf4c · 79baf4c
1 parent 6ae7875
commit 79baf4c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/webnn/validation_tests/conv2d.https.any.js b/webnn/validation_tests/conv2d.https.any.js
@@ -210,6 +210,11 @@ const tests = [
     input: {dataType: 'float32', dimensions: [1, 5, 5]},
     filter: {dataType: 'float32', dimensions: [1, 2, 2, 1]},
   },
+  {
+    name: '[conv2d] Throw if the input data type is not floating point.',
+    input: {dataType: 'int32', dimensions: [1, 1, 5, 5]},
+    filter: {dataType: 'int32', dimensions: [1, 1, 2, 2]},
+  },
   {
     name: '[conv2d] Throw if the filter is not a 4-D tensor.',
     input: {dataType: 'float32', dimensions: [1, 1, 5, 5]},

diff --git a/webnn/validation_tests/convTranspose2d.https.any.js b/webnn/validation_tests/convTranspose2d.https.any.js
@@ -195,6 +195,12 @@ const tests = [
     input: {dataType: 'float32', dimensions: [1, 5, 5]},
     filter: {dataType: 'float32', dimensions: [1, 1, 2, 2]},
   },
+  {
+    name:
+        '[convTranspose2d] Throw if the input data type is not floating point.',
+    input: {dataType: 'int32', dimensions: [1, 1, 5, 5]},
+    filter: {dataType: 'int32', dimensions: [1, 1, 2, 2]},
+  },
   {
     name: '[convTranspose2d] Throw if the filter is not a 4-D tensor.',
     input: {dataType: 'float32', dimensions: [1, 1, 5, 5]},