mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
20 lines
635 B
C++
20 lines
635 B
C++
#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
|