blob: b91e99e289e8ec3dbb7ef37ef189f54316b351c2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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);
|
