Lower memory_kind on inputs/outputs to `mhlo.memory_kind` and then translate that to entry_computation_layout attached to HLO.
Lower memory_kind on inputs/outputs to `mhlo.memory_kind` and then translate that to entry_computation_layout attached to HLO.

This is very similar to how layouts are plumbed from stablehlo to entry_computation_layout.

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/tensorflow/pull/63288 from tensorflow:Surya_QuantizedConv2D_checkfail_fix bba7c37c810dc340530444caf069c463ead6efe3
