[webnn] Add tests for concat operation.

webnn/concat.https.any.js

@@ -0,0 +1,70 @@
+// META: title=test WebNN API concat operation
+// META: global=window,dedicatedworker
+// META: script=./resources/utils.js
+// META: timeout=long
+
+'use strict';
+
+// https://webmachinelearning.github.io/webnn/#api-mlgraphbuilder-concat
+
+(async () => {
+  const testConcat = async (operandType, syncFlag, inputShapeValues, axis, expectedShapeValue) => {
+    const inputs = [];
+    const namedInputs = {};
+    const TestTypedArray = TypedArrayDict[operandType];
+    for (let i = 0; i < inputShapeValues.length; i++) {
+      inputs.push(builder.input('input' + i, {type: operandType, dimensions: inputShapeValues[i].shape}));
+      namedInputs['input' + i] = new TestTypedArray(inputShapeValues[i].data);
+    }
+    const outputOperand = builder.concat(inputs, axis);
+    const outputs = {'outputOperand': new TestTypedArray(sizeOfShape(expectedShapeValue.shape))};
+    let graph;
+    if (syncFlag) {
+      graph = builder.build({outputOperand});
+      context.compute(graph, namedInputs, outputs);
+    } else {
+      graph = await builder.buildAsync({outputOperand});
+      await context.computeAsync(graph, namedInputs, outputs);
+    }
+    assert_array_approx_equals_ulp(outputs.outputOperand, expectedShapeValue.data, PrecisionMetrics.concat.ULP[operandType], operandType);
+  };
+
+  const testsDict = await getTestsJsonInfo('/webnn/resources/test_data/concat.json');
+  const tests = testsDict.tests;
+  const inputsData = testsDict.inputsData;
+  const expectedData = testsDict.expectedData;
+  let context;
+  let builder;
+
+  ExecutionArray.forEach(executionType => {
+    const isSync = executionType === 'sync';
+    if (self.GLOBAL.isWindow() && isSync) {
+      return;
+    }
+
+    DeviceTypeArray.forEach(deviceType => {
+      promise_setup(async () => {
+        context = navigator.ml.createContext({deviceType});
+        builder = new MLGraphBuilder(context);
+      });
+
+      for (const test of tests) {
+        const operandType = test.type;
+        promise_test(async () => {
+          const inputShapeValues = [];
+          const inputShapes = test.inputs.shape;
+          const inputDataCategory = test.inputs.data;
+          const expectedDataCategory = test.expected.data;
+          let pos = 0;
+          for (const shape of inputShapes) {
+            const size = sizeOfShape(shape);
+            inputShapeValues.push({shape, data: inputsData[inputDataCategory].slice(pos, pos + size)});
+            pos += size;
+          }
+          const expectedShapeValue = {shape: test.expected.shape, data: expectedData[expectedDataCategory]};
+          await testConcat(operandType, isSync, inputShapeValues, test.axis, expectedShapeValue);
+        }, `test ${test.name} / ${operandType} / ${deviceType} / ${executionType}`);
+      }
+    });
+  });
+})();