"correction for TpuExecute_FreeTpuEmbeddingMemoryAllocations "
…memory

allocations, and stash them inside the runtime while TPUExecute is being invoked. These temporary device memory allocations are cleared at the end of TPUExecute.

PiperOrigin-RevId: 540156433