};

// Rounds the dimension based on the ceil mode.
int RoundDim(float dim, bool ceil_mode) {
  if (ceil_mode) {
    return std::ceil(dim);
  }
  return std::floor(dim);
}

// For H or W, calculate the output dimension for average pool.
int CalculateSpatialOutDim(int in, int k, int p, int s, bool ceil_mode) {
  const float effective_size = in - k + (2 * p);
  int out = RoundDim(effective_size / (float)s, ceil_mode) + 1;