Skip to content
Snippets Groups Projects

New feature, bug fixes, code modifications to create a CPP export of Resnet18 (quantified model)

1 unresolved thread
Files
21
@@ -47,8 +47,9 @@ void convolution_forward(
: clamp(CHANNELS_HEIGHT + PADDING_Y - (oy * STRIDE_Y),
0, DILATED_KERNEL_HEIGHT);
const int iy = (oy * STRIDE_Y) - PADDING_Y;
#pragma omp parallel for collapse(2)
#ifdef _OPENMP
#pragma omp parallel for collapse(2)
#endif
for (int ox = 0; ox < OUTPUTS_WIDTH; ++ox) {
for (int output = 0; output < NB_OUTPUTS; ++output) {
// moved to inner loop for collapsing -->
Loading