from tensorflow.python.eager import context as _context

Input:

Output:

Prompt: