Searched refs:parsed_device (Results 1 – 7 of 7) sorted by relevance
64 DeviceNameUtils::ParsedName parsed_device; in ParseShardingFromDevice() local65 if (!DeviceNameUtils::ParseFullName(device_name, &parsed_device)) { in ParseShardingFromDevice()72 } else if (!parsed_device.has_type || !parsed_device.has_id || in ParseShardingFromDevice()73 !absl::StrContains(parsed_device.type, in ParseShardingFromDevice()77 const int core = parsed_device.id; in ParseShardingFromDevice()
547 Device parsed_device; in IsTPUDevice() local549 &parsed_device)) in IsTPUDevice()551 return parsed_device.has_type && parsed_device.type == kDeviceTPU; in IsTPUDevice()555 Device parsed_device; in IsTPUReplicatedCore() local557 &parsed_device)) in IsTPUReplicatedCore()559 return parsed_device.has_type && parsed_device.type == kTPUReplicatedCore; in IsTPUReplicatedCore()
110 tensorflow::DeviceNameUtils::ParsedName parsed_device; in IsSupportedInputOp() local112 &parsed_device)) { in IsSupportedInputOp()115 return parsed_device.has_type && parsed_device.type == kDeviceCPU; in IsSupportedInputOp()
369 tensorflow::DeviceNameUtils::ParsedName parsed_device; in IsSupportedHostInputOp() local371 &parsed_device)) { in IsSupportedHostInputOp()374 return parsed_device.type == "CPU"; in IsSupportedHostInputOp()
64 parsed_device = pydev.DeviceSpec.from_string(device_string)65 parsed_device = parsed_device.replace(device_type="CPU", device_index=0)66 return parsed_device.to_string()
87 DeviceNameUtils::ParsedName parsed_device; in TaskNameFromDeviceName() local88 CHECK(DeviceNameUtils::ParseFullName(device_name, &parsed_device)); in TaskNameFromDeviceName()90 CHECK(DeviceNameUtils::GetTaskName(parsed_device, &task_name)); in TaskNameFromDeviceName()
2566 DeviceNameUtils::ParsedName parsed_device; in InstantiateFunctionsFromSubgraphs() local2567 if (!DeviceNameUtils::ParseFullName(target, &parsed_device)) { in InstantiateFunctionsFromSubgraphs()2571 device_ordinal = parsed_device.id; in InstantiateFunctionsFromSubgraphs()