 {
     const size_t num_out_neurons_current_layer = fp->layers[k]->num_out_neurons;
     const size_t num_in_neurons_current_layer  = fp->layers[k]->num_in_neurons;
-    std::cout << "num_out_neurons_current " << num_out_neurons_current_layer << " num_in_neurons_current " << num_in_neurons_current_layer << std::endl;
+    //std::cout << "num_out_neurons_current " << num_out_neurons_current_layer << " num_in_neurons_current " << num_in_neurons_current_layer << std::endl;
 
     float_type* aux_coeffs;
     float_type* aux_csts;

    const auto end = std::chrono::system_clock::now();

    const std::chrono::duration<double> elapsed_seconds = end - start;
    std::cout << "elapsed time: " << elapsed_seconds.count() << "s" << std::endl << std::endl;
}

void predict_size_of_conv_layer(fppoly_t* fp, size_t& current_size, size_t& last_size, long int& offset_x, long int& offset_y, long int& length_x, long int& length_y, long int& shift_x, long int& shift_y, const size_t k, const size_t num_out_neurons_last_layer)
{
    const size_t num_out_neurons_current_layer = fp->layers[k]->num_out_neurons;
    std::cout << "num_out_neurons_current " << num_out_neurons_current_layer << std::endl;

    offset_x = fp->layers[k]->strides[0]*offset_x - fp->layers[k]->pad[0];
    offset_y = fp->layers[k]->strides[1]*offset_y - fp->layers[k]->pad[1];

    std::cout << "offset_x " << offset_x << " offset_y " << offset_y << std::endl;

    length_x = (length_x - 1)*fp->layers[k]->strides[0] + fp->layers[k]->filter_size[0];
    length_y = (length_y - 1)*fp->layers[k]->strides[1] + fp->layers[k]->filter_size[1];

    std::cout << "length_x " << length_x << " length_y " << length_y << std::endl;

    shift_x = fp->layers[k]->strides[0]*shift_x;
    shift_y = fp->layers[k]->strides[1]*shift_y;

    std::cout << "shift_x " << shift_x << " shift_y " << shift_y << std::endl;

    last_size = current_size;
    current_size = 4*num_out_neurons_last_layer*length_x*length_y*fp->layers[k]->input_size[2]*sizeof(float_type);

    std::cout << "Size: " << current_size << " bytes" << std::endl;

    std::cout << std::endl;
}




    if(num_chunks > 1)
    {
        //std::cout << "Number of chunks " << num_chunks << std::endl;
    }

    return num_chunks;
