如何将2x2表中的div设置为等于行中最高div的高度？

.table-container {
  display: flex;
}

.table-cell {
  display: flex;
  flex-direction: column;
}

.table-cell div {
  flex-grow: 1;
}

var rows = document.getElementsByClassName('table-row');
for (var i = 0; i < rows.length; i++) {
  var divs = rows[i].getElementsByClassName('table-cell');
  var maxHeight = 0;
  for (var j = 0; j < divs.length; j++) {
    maxHeight = Math.max(maxHeight, divs[j].offsetHeight);
  }
  for (var j = 0; j < divs.length; j++) {
    divs[j].style.height = maxHeight + 'px';
  }
}