diff --git a/extension/named_data_map/merged_data_map.cpp b/extension/named_data_map/merged_data_map.cpp index b42701c7587..d1aa0b76b81 100644 --- a/extension/named_data_map/merged_data_map.cpp +++ b/extension/named_data_map/merged_data_map.cpp @@ -59,12 +59,10 @@ namespace executorch::extension { ET_NODISCARD Result MergedDataMap::get_tensor_layout( string_view key) const { const auto it = key_to_map_index_.find(key.data()); - ET_CHECK_OR_RETURN_ERROR( - it != key_to_map_index_.end(), - NotFound, - "Key %s not found in named data maps", - key.data()); - + if (it == key_to_map_index_.end()) { + ET_LOG(Debug, "Key %s not found in named data maps.", key.data()); + return Error::NotFound; + } return named_data_maps_.at(it->second)->get_tensor_layout(key); }