@@ -74,14 +74,14 @@ void build_graph(Tensor& input, Tensor& output, bool comments,
7474 1 , pads, 1 , tmp_values, tmp_bias, impl2);
7575 conv_layer->setName (kConvolution );
7676 layers.push_back (conv_layer);
77- layerpostop.push_back (0 );
77+ layerpostop.push_back (false );
7878 if (comments) std::cout << " ConvLayer added to layers." << std::endl;
7979 }
8080 if (layer_type.find (" relu" ) != std::string::npos) {
8181 auto ew_layer = std::make_shared<EWLayer>(" relu" );
8282 ew_layer->setName (kElementWise );
8383 layers.push_back (ew_layer);
84- layerpostop.push_back (1 );
84+ layerpostop.push_back (true );
8585 if (comments)
8686 std::cout << " Element wise (relu) added to layers" << std::endl;
8787 }
@@ -102,7 +102,7 @@ void build_graph(Tensor& input, Tensor& output, bool comments,
102102 auto fc_layer = std::make_shared<FCLayer>(tensor, tmp_bias);
103103 fc_layer->setName (kFullyConnected );
104104 layers.push_back (fc_layer);
105- layerpostop.push_back (0 );
105+ layerpostop.push_back (false );
106106 if (comments) std::cout << " DenseLayer added to layers." << std::endl;
107107 }
108108
@@ -120,7 +120,7 @@ void build_graph(Tensor& input, Tensor& output, bool comments,
120120 auto pool_layer = std::make_shared<PoolingLayer>(shape, pooltype, impl1);
121121 pool_layer->setName (kPooling );
122122 layers.push_back (pool_layer);
123- layerpostop.push_back (0 );
123+ layerpostop.push_back (false );
124124 if (comments) std::cout << " PoolingLayer added to layers." << std::endl;
125125 }
126126
@@ -129,15 +129,15 @@ void build_graph(Tensor& input, Tensor& output, bool comments,
129129 std::make_shared<FlattenLayer>(std::vector<size_t >({0 , 3 , 2 , 1 }));
130130 flatten_layer->setName (kFlatten );
131131 layers.push_back (flatten_layer);
132- layerpostop.push_back (0 );
132+ layerpostop.push_back (false );
133133 if (comments) std::cout << " FlattenLayer added to layers." << std::endl;
134134 }
135135
136136 if (layer_type.find (" Dropout" ) != std::string::npos) {
137137 auto dropout_layer = std::make_shared<DropOutLayer>(0.0 );
138138 dropout_layer->setName (kDropout );
139139 layers.push_back (dropout_layer);
140- layerpostop.push_back (0 );
140+ layerpostop.push_back (false );
141141 if (comments)
142142 std::cout
143143 << " DropOutLayer added to layers with probability 0.4 (turned "
0 commit comments