diff options
Diffstat (limited to 'testing/web-platform/tests/webnn/gemm.https.any.js')
-rw-r--r-- | testing/web-platform/tests/webnn/gemm.https.any.js | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/testing/web-platform/tests/webnn/gemm.https.any.js b/testing/web-platform/tests/webnn/gemm.https.any.js new file mode 100644 index 0000000000..179a1649b6 --- /dev/null +++ b/testing/web-platform/tests/webnn/gemm.https.any.js @@ -0,0 +1,28 @@ +// META: title=test WebNN API gemm operation +// META: global=window,dedicatedworker +// META: script=./resources/utils.js +// META: timeout=long + +'use strict'; + +// https://webmachinelearning.github.io/webnn/#api-mlgraphbuilder-gemm + +const buildGemm= (operationName, builder, resources) => { + // MLOperand gemm(MLOperand a, MLOperand b, optional MLGemmOptions options = {}); + const namedOutputOperand = {}; + const [inputOperandA, inputOperandB] = createMultiInputOperands(builder, resources); + let gemmOptions = {...resources.options}; + if (gemmOptions.c) { + if (gemmOptions.c.shape) { + gemmOptions.c = createConstantOperand(builder, gemmOptions.c); + } else { + // MLOperand c; + // Create a single-value operand when c is a scalar + gemmOptions.c = builder.constant(gemmOptions.c); + } + } + namedOutputOperand[resources.expected.name] = builder[operationName](inputOperandA, inputOperandB, gemmOptions); + return namedOutputOperand; +}; + +testWebNNOperation('gemm', buildGemm);
\ No newline at end of file |