num_layers));

        // current_layer_index Composite Attribute.
        mlir::StringAttr current_layer_str =
            builder.getStringAttr("layer_index");
        NamedAttribute current_layer_attr(
            current_layer_str,
            IntegerAttr::get(IntegerType::get(fn.getContext(), /*width=*/32),
                             current_layer_index++));