// Slices input attribute of rank two and returns the specified column.
//
// Always returns 64 bit integer attribute regardless of bitwidth of the input
// attribute.
static DenseIntElementsAttr SliceDenseIntElementsAttrColumn2D(
    ElementsAttr input, int column) {
  auto int_attr = mlir::cast<DenseIntElementsAttr>(input);
  auto shaped_type = int_attr.getType();
  auto shape = shaped_type.getShape();