Skip to content

Commit bf969e1

Browse files
author
AndreySorokin7
committed
fix tidy
1 parent 4df2bea commit bf969e1

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

app/Graph/build.cpp

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)