Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dpmap_lib.cc166 PmapCacheEntry* GetCacheEntryIfPresent(const CallSignature& signature);
192 PmapCacheEntry* PmapFunction::GetCacheEntryIfPresent( in GetCacheEntryIfPresent() function in jax::PmapFunction
280 PmapCacheEntry* cache_entry = GetCacheEntryIfPresent(arguments.signature); in Call()
283 cache_entry = GetCacheEntryIfPresent(arguments.signature); in Call()
Djax_jit.cc867 CacheEntry* GetCacheEntryIfPresent(const CallSignature& signature);
1026 CacheEntry* CompiledFunction::GetCacheEntryIfPresent( in GetCacheEntryIfPresent() function in jax::__anone04173081411::CompiledFunction
1158 CacheEntry* cache_entry = GetCacheEntryIfPresent(arguments.signature); in Call()
1162 cache_entry = GetCacheEntryIfPresent(arguments.signature); in Call()