def _d_print(inputs, name: str = 'Foo'):
  """Print shape of inputs, which is tensor or list of tensors."""
  is_list = isinstance(inputs, (list, tuple))
  print('{}:   ({})'.format(name, 'List' if is_list else 'Single'))
  if not is_list:
    print('   ', inputs.dtype, inputs.shape, inputs.name, flush=True)
  else:
    for ten in inputs:
      print('   ', ten.dtype, ten.shape, ten.name, flush=True)
  print('', flush=True)
  return inputs