Searched refs:device_dict (Results 1 – 5 of 5) sorted by relevance
77 device_dict = collections.defaultdict(list)119 device_dict[spec.task].append(spec.device_index)125 [len(core_ids) for core_ids in device_dict.values()])153 num_hosts=len(device_dict),
113 device_dict = {}119 if d_spec.job not in device_dict:120 device_dict[d_spec.job] = []123 while len(device_dict[d_spec.job]) <= d_spec.task:124 device_dict[d_spec.job].append([])126 device_dict[d_spec.job][d_spec.task].append(d)128 return device_dict155 device_dict = _group_device_list(devices)157 for _, devices_in_task in device_dict.items():402 device_dict = _group_device_list(devices)[all …]
139 def _verify_and_return_same_core_count(device_dict): argument142 {len(core_ids) for core_ids in device_dict.values()})145 'should never happen. Devices: {}'.format(device_dict))
590 device_dict, num_cores =\593 self.assertEqual(device_dict, {})
162 auto device_dict = builder.getDictionaryAttr( in TEST() local165 (*module_ref)->setAttr("tf.devices", device_dict); in TEST()