summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/webnn/concat.https.any.js
diff options
context:
space:
mode:
Diffstat (limited to 'testing/web-platform/tests/webnn/concat.https.any.js')
-rw-r--r--testing/web-platform/tests/webnn/concat.https.any.js22
1 files changed, 22 insertions, 0 deletions
diff --git a/testing/web-platform/tests/webnn/concat.https.any.js b/testing/web-platform/tests/webnn/concat.https.any.js
new file mode 100644
index 0000000000..b91e99e289
--- /dev/null
+++ b/testing/web-platform/tests/webnn/concat.https.any.js
@@ -0,0 +1,22 @@
+// 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
+
+const buildConcat = (operationName, builder, resources) => {
+ // MLOperand concat(sequence<MLOperand> inputs, unsigned long axis);
+ const namedOutputOperand = {};
+ const inputOperands = [];
+ for (let input of resources.inputs) {
+ inputOperands.push(builder.input(input.name, {type: input.type, dimensions: input.shape}));
+ }
+ // invoke builder.concat()
+ namedOutputOperand[resources.expected.name] = builder[operationName](inputOperands, resources.axis);
+ return namedOutputOperand;
+};
+
+testWebNNOperation('concat', buildConcat); \ No newline at end of file