mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-17 21:10:43 +00:00
25 lines
804 B
JavaScript
25 lines
804 B
JavaScript
|
// Copyright (c) Microsoft Corporation. All rights reserved.
|
||
|
|
// Licensed under the MIT License.
|
||
|
|
|
||
|
|
'use strict';
|
||
|
|
|
||
|
|
it('Browser E2E testing - WebGPU backend with external data', async function() {
|
||
|
|
const session = await ort.InferenceSession.create('./model_with_orig_ext_data.onnx', {
|
||
|
|
executionProviders: ['webgpu'],
|
||
|
|
externalData: [{data: './model_with_orig_ext_data.bin', path: 'model_with_orig_ext_data.bin'}]
|
||
|
|
});
|
||
|
|
|
||
|
|
const fetches = await session.run({X: new ort.Tensor('float32', [1, 1], [1, 2])});
|
||
|
|
|
||
|
|
const Y = fetches.Y;
|
||
|
|
|
||
|
|
assert(Y instanceof ort.Tensor);
|
||
|
|
assert(Y.dims.length === 2 && Y.dims[0] === 2 && Y.dims[1] === 3);
|
||
|
|
assert(Y.data[0] === 1);
|
||
|
|
assert(Y.data[1] === 1);
|
||
|
|
assert(Y.data[2] === 0);
|
||
|
|
assert(Y.data[3] === 0);
|
||
|
|
assert(Y.data[4] === 0);
|
||
|
|
assert(Y.data[5] === 0);
|
||
|
|
});
|