Searched refs:tensor_info_dict (Results 1 – 1 of 1) sorted by relevance
264 def _get_shapes_from_tensor_info_dict(tensor_info_dict): argument275 for key, tensor_info in tensor_info_dict.items()279 def _get_types_from_tensor_info_dict(tensor_info_dict): argument290 for key, tensor_info in tensor_info_dict.items()