#pragma once
|
|
namespace caffe2 {
|
|
// Get value from map given key. Return suppiled default value if not found
|
// This is a stripped down version from folly:
|
// https://github.com/facebook/folly/blob/5a07e203d79324b68d69f294fa38e43b9671e9b1/folly/MapUtil.h#L35-L45
|
template <
|
class Map,
|
typename Key = typename Map::key_type,
|
typename Value = typename Map::mapped_type>
|
typename Map::mapped_type
|
get_default(const Map& map, const Key& key, Value&& dflt) {
|
using M = typename Map::mapped_type;
|
auto pos = map.find(key);
|
return (pos != map.end()) ? (pos->second) : M(std::forward<Value>(dflt));
|
}
|
|
} // namespace caffe2
|
