Add a context reference to registered host buffer.
Add a context reference to registered host buffer.

This prevents that the context is destroyed before the buffer is unregistered.

PiperOrigin-RevId: 392716903

tensorflow/runtime,tensorflow/runtime,tensorflow/runtime,tensorflow/runtime