return GetTensorFeatureDimIndex(input_ty.getRank(), format);
}

// Gets all integer values from the given attribute and push them to `values`.
void GetI64ArrayAttrValues(Attribute attr, SmallVectorImpl<int64_t> *values) {
  auto array_attr = mlir::cast<ArrayAttr>(attr);
  values->reserve(array_attr.getValue().size());
  for (Attribute val : array_attr.getValue())