diff options
Diffstat (limited to 'testing/web-platform/tests/webnn/concat.https.any.js')
-rw-r--r-- | testing/web-platform/tests/webnn/concat.https.any.js | 22 |
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..2c8950fe99 --- /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, 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', '/webnn/resources/test_data/concat.json', buildConcat);
\ No newline at end of file |