This commit is contained in:
Francesco Gatti
2019-01-04 22:28:10 +01:00
parent 2e8d0b1002
commit 88097a3774
7 changed files with 139 additions and 23 deletions
+1 -1
View File
@@ -47,7 +47,7 @@ dnnType* Activation::infer(dataDim_t &dim, dnnType* srcData) {
if(act_mode == ACTIVATION_LEAKY) {
activationLEAKYForward(srcData, dstData, dim.tot());
} else {
dnnType alpha = dnnType(1);
dnnType beta = dnnType(0);