from tensorflow.python.framework import dtypes as _dtypes

Input:

Output:

Prompt: