webnn: Enforce input data type constraints for resample2d

lisa0314 · chromium-wpt-export-bot · commit b5ab7ede813b · 2024-05-07T01:34:18.000-07:00
As specified in webmachinelearning/webnn#646 Bug: 328567884 Change-Id: Icea0e4a586aee33d7993d6e72393412b981d26ab Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/5495876 Reviewed-by: Austin Sullivan <asully@chromium.org> Reviewed-by: ningxin hu <ningxin.hu@intel.com> Commit-Queue: Lisha Guo <lisha.guo@intel.com> Cr-Commit-Position: refs/heads/main@{#1297332}
diff --git a/webnn/validation_tests/resample2d.https.any.js b/webnn/validation_tests/resample2d.https.any.js
@@ -23,6 +23,13 @@ const tests = [
     options: {scales: [0.5, 0.5]},
     output: {dataType: 'float32', dimensions: [1, 1, 2, 2]},
   },
+  {
+    name:
+        '[resample2d] Test building resample2d with input\'s dataType = float16',
+    input: {dataType: 'float16', dimensions: [1, 1, 5, 5]},
+    options: {scales: [0.5, 0.5]},
+    output: {dataType: 'float16', dimensions: [1, 1, 2, 2]},
+  },
   {
     name:
         '[resample2d] Test building resample2d with scales=[0.5, 0.5] and explicit axes=[2, 3]',
@@ -51,6 +58,11 @@ const tests = [
     options: {scales: [2.0, 2.0], sizes: [3, 6]},
     output: {dataType: 'float32', dimensions: [1, 1, 3, 6]},
   },
+  {
+    name:
+        '[resample2d] Throw if the dataType of input is not float32 or float16',
+    input: {dataType: 'int32', dimensions: [2, 4]},
+  },
   {
     name: '[resample2d] Throw if the rank of input is not 4',
     input: {dataType: 'float32', dimensions: [2, 4]},
@@ -75,6 +87,11 @@ const tests = [
     input: {dataType: 'float32', dimensions: [1, 1, 2, 4]},
     options: {sizes: [1, 1, 4, 6]},
   },
+  {
+    name: '[resample2d] Throw if input data type is not floating type',
+    input: {dataType: 'int32', dimensions: [1, 1, 2, 4]},
+    options: {sizes: [1, 1, 4, 6]},
+  },
   {
     name:
         '[resample2d] Throw if any size value is out of \'unsigned long\' value range',
