From 897226c0f3728b8f98e96dd01b5b7cc42f65c6dc Mon Sep 17 00:00:00 2001 From: Wanming Lin Date: Tue, 11 Jun 2024 09:55:15 +0800 Subject: [PATCH] Unfuse activation in facenet_nchw.js --- face_recognition/facenet_nchw.js | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/face_recognition/facenet_nchw.js b/face_recognition/facenet_nchw.js index bbb6a1ff..bc19c9dc 100644 --- a/face_recognition/facenet_nchw.js +++ b/face_recognition/facenet_nchw.js @@ -41,9 +41,7 @@ export class FaceNetNchw { bias: bias, }; } - if (relu) { - options.activation = this.builder_.relu(); - } + // WebNN spec drops autoPad support, compute the explicit padding instead. if (options.autoPad == 'same-upper') { options.padding = @@ -52,7 +50,8 @@ export class FaceNetNchw { /* oihw */[weights.shape()[2], weights.shape()[3]], options.strides, options.dilations, options.autoPad); } - return this.builder_.conv2d(input, weights, options); + const conv2d = this.builder_.conv2d(input, weights, options); + return relu ? this.builder_.relu(conv2d) : conv2d; } async buildBlock35_(input, indice, weightsSuffixes) {