Searched refs:parsed_device (Results 1 – 7 of 7) sorted by relevance
63 DeviceNameUtils::ParsedName parsed_device; in ParseShardingFromDevice() local64 if (!DeviceNameUtils::ParseFullName(device_name, &parsed_device)) { in ParseShardingFromDevice()71 } else if (!parsed_device.has_type || !parsed_device.has_id || in ParseShardingFromDevice()72 !absl::StrContains(parsed_device.type, in ParseShardingFromDevice()76 const int core = parsed_device.id; in ParseShardingFromDevice()
127 tensorflow::DeviceNameUtils::ParsedName parsed_device; in IsSupportedInputOp() local129 &parsed_device)) { in IsSupportedInputOp()132 return parsed_device.has_type && parsed_device.type == kDeviceCPU; in IsSupportedInputOp()
414 tensorflow::DeviceNameUtils::ParsedName parsed_device; in IsSupportedHostInputOp() local416 &parsed_device)) { in IsSupportedHostInputOp()419 return parsed_device.type == "CPU"; in IsSupportedHostInputOp()
65 parsed_device = pydev.DeviceSpec.from_string(device_string)66 parsed_device = parsed_device.replace(device_type="CPU", device_index=0)67 return parsed_device.to_string()
545 Device parsed_device; in IsTPUDevice() local547 &parsed_device)) in IsTPUDevice()549 return parsed_device.has_type && parsed_device.type == kDeviceTPU; in IsTPUDevice()
79 DeviceNameUtils::ParsedName parsed_device; in TaskNameFromDeviceName() local80 CHECK(DeviceNameUtils::ParseFullName(device_name, &parsed_device)); in TaskNameFromDeviceName()82 CHECK(DeviceNameUtils::GetTaskName(parsed_device, &task_name)); in TaskNameFromDeviceName()
2805 DeviceNameUtils::ParsedName parsed_device; in CreateOrGetPerHostVariableCopy() local2806 TF_RET_CHECK(DeviceNameUtils::ParseFullName(host_cpu_device, &parsed_device)); in CreateOrGetPerHostVariableCopy()2807 TF_RET_CHECK(parsed_device.has_task); in CreateOrGetPerHostVariableCopy()2808 VLOG(1) << "Creating per-host IdentityN node for task " << parsed_device.task; in CreateOrGetPerHostVariableCopy()2846 if (parsed_device.task == 0 || !enable_xla_param_broadcast) { in CreateOrGetPerHostVariableCopy()2869 absl::StrFormat("/dummy_%d", parsed_device.task)); in CreateOrGetPerHostVariableCopy()