We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 919aa9f + 897226c commit b81e927Copy full SHA for b81e927
face_recognition/facenet_nchw.js
@@ -41,9 +41,7 @@ export class FaceNetNchw {
41
bias: bias,
42
};
43
}
44
- if (relu) {
45
- options.activation = this.builder_.relu();
46
- }
+
47
// WebNN spec drops autoPad support, compute the explicit padding instead.
48
if (options.autoPad == 'same-upper') {
49
options.padding =
@@ -52,7 +50,8 @@ export class FaceNetNchw {
52
50
/* oihw */[weights.shape()[2], weights.shape()[3]],
53
51
options.strides, options.dilations, options.autoPad);
54
55
- return this.builder_.conv2d(input, weights, options);
+ const conv2d = this.builder_.conv2d(input, weights, options);
+ return relu ? this.builder_.relu(conv2d) : conv2d;
56
57
58
async buildBlock35_(input, indice, weightsSuffixes) {
0 commit comments