From 7e6c28f901edb49f900d2d721aedd0040c017604 Mon Sep 17 00:00:00 2001 From: Juhyun Lee Date: Wed, 14 Oct 2020 15:06:20 -0700 Subject: [PATCH] Downgrade c++17 syntax to c++14. PiperOrigin-RevId: 337181092 Change-Id: Id89bc78f3ad1433176184b269e4d659b41c84bcc --- tensorflow/lite/delegates/gpu/cl/arguments.cc | 28 ++++++++++++++----- tensorflow/lite/delegates/gpu/cl/arguments.h | 3 ++ 2 files changed, 24 insertions(+), 7 deletions(-) diff --git a/tensorflow/lite/delegates/gpu/cl/arguments.cc b/tensorflow/lite/delegates/gpu/cl/arguments.cc index 8390112c02d..7c5e635816e 100644 --- a/tensorflow/lite/delegates/gpu/cl/arguments.cc +++ b/tensorflow/lite/delegates/gpu/cl/arguments.cc @@ -756,24 +756,38 @@ void Arguments::ResolveObjectNames(const std::string& object_name, } } +GPUObjectDescriptor* Arguments::GetObjectDescriptor( + const std::string& object_name) const { + { + auto it = object_refs_.find(object_name); + if (it != object_refs_.end()) { + return it->second.descriptor.get(); + } + } + { + auto it = objects_.find(object_name); + if (it != objects_.end()) { + return it->second.descriptor.get(); + } + } + return nullptr; +} + absl::Status Arguments::ResolveSelector( const std::map& linkables, const std::string& object_name, const std::string& selector, const std::vector& args, const std::vector& template_args, std::string* result) { - const GPUObjectDescriptor* desc_ptr; - if (auto it = object_refs_.find(object_name); it != object_refs_.end()) { - desc_ptr = it->second.descriptor.get(); - } else if (auto it = objects_.find(object_name); it != objects_.end()) { - desc_ptr = it->second.descriptor.get(); - } else { + const GPUObjectDescriptor* desc_ptr = GetObjectDescriptor(object_name); + if (!desc_ptr) { return absl::NotFoundError( absl::StrCat("No object with name - ", object_name)); } auto names = desc_ptr->GetGPUResources().GetNames(); const auto* tensor_desc = dynamic_cast(desc_ptr); if (tensor_desc && selector == "Write") { - if (auto it = linkables.find(object_name); it != linkables.end()) { + auto it = linkables.find(object_name); + if (it != linkables.end()) { if (desc_ptr->GetAccess() != AccessType::WRITE && desc_ptr->GetAccess() != AccessType::READ_WRITE) { return absl::FailedPreconditionError(absl::StrCat( diff --git a/tensorflow/lite/delegates/gpu/cl/arguments.h b/tensorflow/lite/delegates/gpu/cl/arguments.h index 2f87f8ae6a0..bac6afd55b7 100644 --- a/tensorflow/lite/delegates/gpu/cl/arguments.h +++ b/tensorflow/lite/delegates/gpu/cl/arguments.h @@ -119,6 +119,9 @@ class Arguments : public ArgumentsBinder { const std::vector& member_names, std::string* code); + GPUObjectDescriptor* GetObjectDescriptor( + const std::string& object_name) const; + static constexpr char kArgsPrefix[] = "args."; struct IntValue {