[TF-M] Handling the SAM partial write