from tensorflow.python.framework import common_shapes as _common_shapes

Input:

Output:

Prompt: