Searched refs:GetDeviceParsedName (Results 1 – 5 of 5) sorted by relevance
369 if (op->GetDeviceParsedName().type == "TPU" || in MustCompileWithXLA()370 op->GetDeviceParsedName().type == "XLA_GPU" || in MustCompileWithXLA()371 op->GetDeviceParsedName().type == "XLA_CPU") { in MustCompileWithXLA()374 << op->GetDeviceParsedName().type; in MustCompileWithXLA()490 ctx.SelectDevice(op->GetDeviceParsedName(), ndef, &device)); in GetOrCreateKernelAndDevice()834 TF_RETURN_IF_ERROR(ctx.GetClient(op->GetDeviceParsedName(), &eager_client)); in EagerRemoteExecute()836 if (!DeviceNameUtils::GetTaskName(op->GetDeviceParsedName(), &remote_task)) { in EagerRemoteExecute()968 DeviceNameUtils::ParsedNameToString(op->GetDeviceParsedName())); in EagerRemoteExecute()
60 const DeviceNameUtils::ParsedName& GetDeviceParsedName() const { in GetDeviceParsedName() function
191 absl::optional<DeviceNameUtils::ParsedName> GetDeviceParsedName(
269 absl::optional<DeviceNameUtils::ParsedName> GetDeviceParsedName( in GetDeviceParsedName() function
747 GetDeviceParsedName(node->tf_node()); in SegmentGraph()