add support for relu6 (#34)

dgschwend · Jul 5, 2018 · 06756ce · 06756ce
1 parent fc06e1c
commit 06756ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/analyzer.coffee b/src/analyzer.coffee
@@ -223,7 +223,7 @@ module.exports =
                     d.mem.activation = d.wOut*d.hOut*d.chOut*d.batchOut
 
                 #relu/dropout use some memory, do some comparisons
-                when "relu", "elu", "prelu", "dropout"
+                when "relu", "relu6", "elu", "prelu", "dropout"
                     #dimensions
                     d.wIn = parent.wOut
                     d.hIn = parent.hOut