If context is None, make a call to context(), to force create it.

enable_eager_execution() guarantees the existence of the context, but in v2, we
will not have that explicit call.

Its faster to access the context directly as opposed to making a function
call, and I don't want to add to the overhead of every single op.

PiperOrigin-RevId: 231852561