Searched refs:GetOrCreateIfAbsent (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | refcounting_hash_map_test.cc | 53 std::shared_ptr<int> a = m.GetOrCreateIfAbsent(0, factory); in TEST()
54 std::shared_ptr<int> b = m.GetOrCreateIfAbsent(0, factory); in TEST()
55 std::shared_ptr<int> c = m.GetOrCreateIfAbsent(1, factory); in TEST()
63 EXPECT_EQ(*m.GetOrCreateIfAbsent(42, factory), 0); in TEST()
70 auto handle = m.GetOrCreateIfAbsent(0, factory); in TEST()
80 EXPECT_EQ(*m.GetOrCreateIfAbsent(0, factory), 1); in TEST()
81 EXPECT_EQ(*m.GetOrCreateIfAbsent(100, factory), 101); in TEST()
|
| D | refcounting_hash_map.h | 59 std::shared_ptr<V> GetOrCreateIfAbsent( in GetOrCreateIfAbsent() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 652 return GlobalAllToAllRendezvousMap().GetOrCreateIfAbsent( in __xla_cpu_runtime_AllToAll()
702 return GlobalAllReduceRendezvousMap().GetOrCreateIfAbsent( in __xla_cpu_runtime_AllReduce()
759 return GlobalCollectivePermuteRendezvousMap().GetOrCreateIfAbsent( in __xla_cpu_runtime_CollectivePermute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | collective_permute_thunk.cc | 241 GlobalRendezvousMap().GetOrCreateIfAbsent(key, rendezvous_factory); in ExecuteOnStream()
|
| D | nccl_utils.cc | 345 return rendezvous_map.GetOrCreateIfAbsent( in AcquireNcclClique()
|