Remove the final parameter from the layers

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-04-09 18:48:03 +02:00
parent 3502c5b676
commit 4361d5fec0
14 changed files with 130 additions and 115 deletions
@@ -5,7 +5,7 @@

 namespace tk { namespace dnn {

-Reshape::Reshape(Network *net, dataDim_t new_dim, bool final) : Layer(net, final) {
+Reshape::Reshape(Network *net, dataDim_t new_dim) : Layer(net) {

    checkCuda( cudaMalloc(&dstData, input_dim.tot()*sizeof(dnnType)) );