Searched refs:parsed_device (Results 1 – 3 of 3) sorted by relevance
58 DeviceNameUtils::ParsedName parsed_device; in ParseShardingFromDevice() local59 if (!DeviceNameUtils::ParseFullName(device_name, &parsed_device)) { in ParseShardingFromDevice()66 } else if (!parsed_device.has_type || !parsed_device.has_id || in ParseShardingFromDevice()67 !absl::StrContains(parsed_device.type, in ParseShardingFromDevice()71 const int core = parsed_device.id; in ParseShardingFromDevice()
52 parsed_device = pydev.DeviceSpec.from_string(device_string)53 parsed_device.device_type = "CPU"54 parsed_device.device_index = 055 return parsed_device.to_string()
110 DeviceNameUtils::ParsedName parsed_device; in CompleteGroupLocal() local111 DeviceNameUtils::ParseFullName(device, &parsed_device); in CompleteGroupLocal()112 string task_name = strings::StrCat("/job:", parsed_device.job, in CompleteGroupLocal()113 "/replica:", parsed_device.replica, in CompleteGroupLocal()114 "/task:", parsed_device.task); in CompleteGroupLocal()