Home
last modified time | relevance | path

Searched refs:PmapCacheEntry (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpmap_lib.cc101 struct PmapCacheEntry { struct
166 PmapCacheEntry* GetCacheEntryIfPresent(const CallSignature& signature);
168 PmapCacheEntry* AddCacheEntry(const py::args& args, const py::kwargs& kwargs,
182 absl::flat_hash_map<CallSignature, std::unique_ptr<PmapCacheEntry>>
192 PmapCacheEntry* PmapFunction::GetCacheEntryIfPresent( in GetCacheEntryIfPresent()
209 PmapCacheEntry* PmapFunction::AddCacheEntry(const py::args& args, in AddCacheEntry()
215 executables_.emplace(signature, std::make_unique<PmapCacheEntry>()); in AddCacheEntry()
217 PmapCacheEntry* cache_entry = it->second.get(); in AddCacheEntry()
280 PmapCacheEntry* cache_entry = GetCacheEntryIfPresent(arguments.signature); in Call()