return row * pool.w_out + col;
  };

  std::vector<float> correction_data(out_shaped_type.getNumElements(), 1.0);

  const float kern_size = pool.kh * pool.kw;

  // LEMMA 1: Changing the rounding mode from floor to ceil will increase an
  // output dimension by at most 1 (see `ComputeSpatialOutDim`). This is because
  // for any `x`, `ceil(x) - floor(x) <= 1`.