Skip to content

Commit a78d45e

Browse files
authored
Update webnn delegate for selfie segmentation sample (#187)
- Update deviceType enum - Add a duplicate entry to support new dataType - Support WebNN numThreads option
1 parent b138e20 commit a78d45e

9 files changed

Lines changed: 5 additions & 2 deletions

β€Žselfie_segmentation/builtin_delegate_worker.jsβ€Ž

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@ onmessage = async (message) => {
3737
numThreads: 1,
3838
enableWebNNDelegate: message.data.enableWebNNDelegate,
3939
webNNDevicePreference: parseInt(message.data.webNNDevicePreference),
40+
webNNNumThreads: parseInt(message.data.webNNNumThreads),
4041
},
4142
);
4243

β€Žselfie_segmentation/main.jsβ€Ž

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -209,6 +209,7 @@ export async function main() {
209209
ui.handleClick(disabledSelectors, true);
210210
if (isFirstTimeLoad) $('#hint').hide();
211211
const numRuns = utils.getUrlParams()[0];
212+
const numThreads = utils.getUrlParams()[2];
212213
// Only do load() when model first time loads and
213214
// there's new model or delegate choosed
214215
if (isFirstTimeLoad || modelChanged) {
@@ -223,6 +224,7 @@ export async function main() {
223224
modelPath: modelConfigs[modelName].modelPath,
224225
enableWebNNDelegate: enableWebnnDelegate,
225226
webNNDevicePreference: 0,
227+
webNNNumThreads: numThreads,
226228
};
227229
loadTime = await postAndListenMessage(options);
228230
console.log(` done in ${loadTime} ms.`);

β€Žselfie_segmentation/tflite-support/tflite_model_runner_cc.jsβ€Ž

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.
Binary file not shown.
Binary file not shown.

β€Žselfie_segmentation/tflite-support/tflite_model_runner_cc_threaded_simd.jsβ€Ž

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.
Binary file not shown.
Binary file not shown.
15.4 KB
Binary file not shown.

0 commit comments

Comments
Β (0)