summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/webnn/split.https.any.js
diff options
context:
space:
mode:
Diffstat (limited to 'testing/web-platform/tests/webnn/split.https.any.js')
-rw-r--r--testing/web-platform/tests/webnn/split.https.any.js24
1 files changed, 24 insertions, 0 deletions
diff --git a/testing/web-platform/tests/webnn/split.https.any.js b/testing/web-platform/tests/webnn/split.https.any.js
new file mode 100644
index 0000000000..bb1c995658
--- /dev/null
+++ b/testing/web-platform/tests/webnn/split.https.any.js
@@ -0,0 +1,24 @@
+// META: title=test WebNN API split operation
+// META: global=window,dedicatedworker
+// META: script=./resources/utils.js
+// META: timeout=long
+
+'use strict';
+
+// https://webmachinelearning.github.io/webnn/#api-mlgraphbuilder-split
+
+const buildSplit = (operationName, builder, resources) => {
+ // sequence<MLOperand> split(MLOperand input,
+ // (unsigned long or sequence<unsigned long>) splits,
+ // optional MLSplitOptions options = {});
+ const namedOutputOperand = {};
+ const inputOperand = createSingleInputOperand(builder, resources);
+ // invoke builder.split()
+ const outputOperands = builder[operationName](inputOperand, resources.splits, resources.options);
+ resources.expected.forEach((resourceDict, index) => {
+ namedOutputOperand[resourceDict.name] = outputOperands[index];
+ });
+ return namedOutputOperand;
+};
+
+testWebNNOperation('split', buildSplit); \ No newline at end of file