File tree Expand file tree Collapse file tree 2 files changed +4
-6
lines changed Expand file tree Collapse file tree 2 files changed +4
-6
lines changed Original file line number Diff line number Diff line change @@ -349,7 +349,7 @@ class Graph {
349
349
return traversal;
350
350
}
351
351
352
- std::string layerTypeToString (it_lab_ai::LayerType type) {
352
+ std::string layerTypeToString (it_lab_ai::LayerType type) {
353
353
switch (type) {
354
354
case it_lab_ai::kInput :
355
355
return " Input" ;
@@ -387,12 +387,12 @@ std::string layerTypeToString(it_lab_ai::LayerType type) {
387
387
return " Unknown" ;
388
388
}
389
389
}
390
- std::string getLayerName (int layer_index) {
390
+ std::string getLayerName (int layer_index) {
391
391
if (layer_index >= 0 && layer_index < static_cast <int >(layers_.size ())) {
392
392
it_lab_ai::LayerType type = layers_[layer_index]->getName ();
393
393
return layerTypeToString (type);
394
394
}
395
395
return " Unknown_Layer" ;
396
- }
396
+ }
397
397
};
398
398
} // namespace it_lab_ai
Original file line number Diff line number Diff line change @@ -167,9 +167,7 @@ PoolingLayerImpl<ValueType>::PoolingLayerImpl(
167
167
input_shape[input_shape.dims () - pooling_shape.dims () + i];
168
168
size_t kernel_size = pooling_shape[i];
169
169
size_t stride = strides[i];
170
- size_t pad =
171
- pads[i] +
172
- pads[pooling_shape.dims () + i];
170
+ size_t pad = pads[i] + pads[pooling_shape.dims () + i];
173
171
size_t dilation = dilations[i];
174
172
175
173
size_t effective_kernel_size = (kernel_size - 1 ) * dilation + 1 ;
You can’t perform that action at this time.
0 commit comments