Update webnn delegate for selfie segmentation sample (#187)

- Update deviceType enum - Add a duplicate entry to support new dataType - Support WebNN numThreads option
webmachinelearning · Nov 3, 2023 · a78d45e · a78d45e
1 parent b138e20
commit a78d45e
Show file tree

Hide file tree

Showing 9 changed files with 5 additions and 2 deletions.
diff --git a/selfie_segmentation/builtin_delegate_worker.js b/selfie_segmentation/builtin_delegate_worker.js
@@ -37,6 +37,7 @@ onmessage = async (message) => {
                 numThreads: 1,
                 enableWebNNDelegate: message.data.enableWebNNDelegate,
                 webNNDevicePreference: parseInt(message.data.webNNDevicePreference),
+                webNNNumThreads: parseInt(message.data.webNNNumThreads),
               },
           );
 

diff --git a/selfie_segmentation/main.js b/selfie_segmentation/main.js
@@ -209,6 +209,7 @@ export async function main() {
     ui.handleClick(disabledSelectors, true);
     if (isFirstTimeLoad) $('#hint').hide();
     const numRuns = utils.getUrlParams()[0];
+    const numThreads = utils.getUrlParams()[2];
     // Only do load() when model first time loads and
     // there's new model or delegate choosed
     if (isFirstTimeLoad || modelChanged) {
@@ -223,6 +224,7 @@ export async function main() {
         modelPath: modelConfigs[modelName].modelPath,
         enableWebNNDelegate: enableWebnnDelegate,
         webNNDevicePreference: 0,
+        webNNNumThreads: numThreads,
       };
       loadTime = await postAndListenMessage(options);
       console.log(`  done in ${loadTime} ms.`);

diff --git a/selfie_segmentation/tflite-support/tflite_model_runner_cc.js b/selfie_segmentation/tflite-support/tflite_model_runner_cc.js
diff --git a/selfie_segmentation/tflite-support/tflite_model_runner_cc.wasm b/selfie_segmentation/tflite-support/tflite_model_runner_cc.wasm
diff --git a/selfie_segmentation/tflite-support/tflite_model_runner_cc_simd.wasm b/selfie_segmentation/tflite-support/tflite_model_runner_cc_simd.wasm
diff --git a/selfie_segmentation/tflite-support/tflite_model_runner_cc_threaded_simd.js b/selfie_segmentation/tflite-support/tflite_model_runner_cc_threaded_simd.js
diff --git a/selfie_segmentation/tflite-support/tflite_model_runner_cc_threaded_simd.wasm b/selfie_segmentation/tflite-support/tflite_model_runner_cc_threaded_simd.wasm
diff --git a/selfie_segmentation/tflite-support/webnn_external_delegate_wasm.wasm b/selfie_segmentation/tflite-support/webnn_external_delegate_wasm.wasm
diff --git a/selfie_segmentation/tflite_model_runner_cc_simd.wasm b/selfie_segmentation/tflite_model_runner_cc_simd.wasm