correction_data[flat_ind] = kern_size / correct_divisor;
    }
  }

  return DenseFPElementsAttr::get(out_shaped_type, correction_data);
}

std::array<int, 4> GetPadOpPaddingValues(const TorchAvgPoolData& pool) {
  int pad_bottom = pool.ph;
  int pad_right = pool.pw;

  if (pool.ceil_mode) {
    const int remaining_bottom = pool.h_in - ((pool.h_out - 1) * pool.sh);