Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compilation_cache_rpc_lookup.cc73 std::vector<std::shared_ptr<CacheEntry>> removed_entries; in Lookup() local
93 PostLookupLocked(&cache_entry, entry, &removed_entries); in Lookup()
108 std::vector<std::shared_ptr<CacheEntry>> removed_entries; in Lookup() local
138 PostLookupLocked(&cache_entry, entry, &removed_entries); in Lookup()
173 std::vector<std::shared_ptr<CacheEntry>>* removed_entries) { in PostLookupLocked() argument
192 removed_entries->push_back(entry_to_evict_it->second); in PostLookupLocked()
Dtpu_compilation_cache_interface.cc283 std::vector<CompiledSubgraph*> removed_entries; in DiscardEntryRefs() local
288 removed_entries.push_back(DiscardEntryRef(entry)); in DiscardEntryRefs()
297 for (auto removed_entry : removed_entries) { in DiscardEntryRefs()
319 CompiledSubgraph* entry, std::vector<CompiledSubgraph*>* removed_entries) { in LookupEntryMarkedForEviction() argument
334 removed_entries->push_back(entry_to_evict); in LookupEntryMarkedForEviction()
369 std::vector<CompiledSubgraph*> removed_entries; in CompileIfKeyAbsent() local
372 sharding_key, may_modify_variables, &removed_entries, hlo_metadatas, in CompileIfKeyAbsent()
374 for (auto entry : removed_entries) { in CompileIfKeyAbsent()
405 std::vector<CompiledSubgraph*>* removed_entries, in CompileIfKeyAbsentHelper() argument
527 LookupEntryMarkedForEviction(entry, removed_entries); in CompileIfKeyAbsentHelper()
Dtpu_compilation_cache_interface.h203 std::vector<CompiledSubgraph*>* removed_entries,
243 CompiledSubgraph* entry, std::vector<CompiledSubgraph*>* removed_entries)
Dtpu_compilation_cache_rpc_lookup.h67 std::vector<std::shared_ptr<CacheEntry>>* removed_entries)