[webnn] Add float32 tests for WebNN API expand op (#43559)

web-platform-tests · Dec 13, 2023 · b597a15 · b597a15
1 parent c849fb6
commit b597a15
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 2 deletions.
diff --git a/webnn/expand.https.any.js b/webnn/expand.https.any.js
@@ -0,0 +1,11 @@
+// META: title=test WebNN API expand operation
+// META: global=window,dedicatedworker
+// META: script=./resources/utils.js
+// META: timeout=long
+
+'use strict';
+
+// https://webmachinelearning.github.io/webnn/#api-mlgraphbuilder-expand
+
+// reuse buildReshape method
+testWebNNOperation('expand', buildReshape);
diff --git a/webnn/gpu/expand.https.any.js b/webnn/gpu/expand.https.any.js
@@ -0,0 +1,11 @@
+// META: title=test WebNN API expand operation
+// META: global=window,dedicatedworker
+// META: script=../resources/utils.js
+// META: timeout=long
+
+'use strict';
+
+// https://webmachinelearning.github.io/webnn/#api-mlgraphbuilder-expand
+
+// reuse buildReshape method
+testWebNNOperation('expand', buildReshape, 'gpu');
diff --git a/webnn/resources/utils.js b/webnn/resources/utils.js
@@ -318,6 +318,7 @@ const PrecisionMetrics = {
   tan: {ATOL: {float32: 1/1024, float16: 1/512}},
   // End Element-wise unary operations
   elu: {ULP: {float32: 18, float16: 18}},
+  expand: {ULP: {float32: 0, float16: 0}},
   gemm: {ULP: {float32: getGemmPrecisionTolerance, float16: getGemmPrecisionTolerance}},
   hardSigmoid: {ULP: {float32: 2, float16: 2}},
   hardSwish: {ULP: {float32: 4, float16: 4}},
@@ -661,10 +662,11 @@ const buildPad = (operationName, builder, resources) => {
 };
 
 const buildReshape = (operationName, builder, resources) => {
-  // MLOperand reshape(MLOperand input, sequence<unsigned long?> newShape);
+  // MLOperand reshape(MLOperand input, sequence<unsigned long> newShape);
+  // MLOperand expand(MLOperand input, sequence<unsigned long> newShape);
   const namedOutputOperand = {};
   const inputOperand = createSingleInputOperand(builder, resources);
-  // invoke builder.reshape()
+  // invoke builder.reshape() or builder.expand()
   namedOutputOperand[resources.expected.name] = builder[operationName](inputOperand, resources.newShape);
   return namedOutputOperand;
 };