Searched refs:GetDeviceParsedName (Results 1 – 2 of 2) sorted by relevance
319 if (op->GetDeviceParsedName().type == "TPU" || in MustCompileWithXLA()320 op->GetDeviceParsedName().type == "XLA_GPU" || in MustCompileWithXLA()321 op->GetDeviceParsedName().type == "XLA_CPU") { in MustCompileWithXLA()324 << op->GetDeviceParsedName().type; in MustCompileWithXLA()461 TF_RETURN_IF_ERROR(ctx.SelectDevice(op->GetDeviceParsedName(), in EagerLocalExecute()642 TF_RETURN_IF_ERROR(ctx.GetClient(op->GetDeviceParsedName(), &eager_client)); in EagerRemoteExecute()644 if (!DeviceNameUtils::GetTaskName(op->GetDeviceParsedName(), &remote_task)) { in EagerRemoteExecute()926 DeviceNameUtils::ParsedNameToString(op->GetDeviceParsedName())); in EagerExecute()
87 const DeviceNameUtils::ParsedName& GetDeviceParsedName() const { in GetDeviceParsedName() function