kernel_spatial_dimensions.resize(num_spatial_dims);
  std::iota(kernel_spatial_dimensions.begin(), kernel_spatial_dimensions.end(),
            0);

  return builder->getNamedAttr(
      "dimension_numbers",
      ConvDimensionNumbersAttr::get(
          builder->getContext(), batch_dim, feature_dim, spatial_dims,
          kernel_input_feature_dim, kernel_output_feature_dim,

    DefaultValuedOptionalAttr<I64Attr, "1">:$batch_group_count,
    DefaultValuedOptionalAttr<I64Attr, "1">:$feature_group_count,
    DefaultValuedOptionalAttr<StrAttr, "\"\"">:$dimension_numbers,
    DefaultValuedOptionalAttr<I64Attr, "-1">:$lhs_quantization_axis,
    I64Attr:$lhs_quantization_min_val,
    I64Attr:$lhs_quantization_max_val,