jax

082f715c - Preserve existing mhlo.frontend_attributes when adding compute_type annotations.

Preserve existing mhlo.frontend_attributes when adding compute_type annotations. The wrap_compute_type_in_place function now merges the compute type related attributes with any preexisting attributes within the mhlo.frontend_attributes dictionary, rather than overwriting the entire dictionary. This ensures that other frontend attributes are not lost. PiperOrigin-RevId: 885739571