Searched refs:dict_or_tensor (Results 1 – 1 of 1) sorted by relevance
407 def _get_dict(name, dict_or_tensor, key): argument410 if not isinstance(dict_or_tensor, dict):413 name + 's dict, got %s.\n' % dict_or_tensor +416 if key not in dict_or_tensor:418 'Key \'%s\' missing from %s.' % (key, dict_or_tensor.keys()))419 return dict_or_tensor[key]421 if isinstance(dict_or_tensor, dict):422 if len(dict_or_tensor) != 1:425 ' dict, got %s' % dict_or_tensor)426 return six.next(six.itervalues(dict_or_tensor))[all …]