Searched refs:tensors_dict (Results 1 – 1 of 1) sorted by relevance
799 def do_one(tensors_dict, dim): argument800 for category, tensors in tensors_dict.items():