Searched refs:jit_executable_cache (Results 1 – 5 of 5) sorted by relevance
26 tfrt::cpu::jit::JitExecutableCache* jit_executable_cache) in TfCpuRtRequestState()27 : jit_executable_cache(jit_executable_cache) {} in TfCpuRtRequestState()30 tfrt::cpu::jit::JitExecutableCache* jit_executable_cache; member
29 explicit TfJitRtRequestState(JitExecutableCache* jit_executable_cache) in TfJitRtRequestState()30 : jit_executable_cache(jit_executable_cache) {} in TfJitRtRequestState()33 JitExecutableCache* jit_executable_cache; member
295 JitExecutableCache* jit_executable_cache = state->jit_executable_cache; in CompileImpl() local298 auto cached = jit_executable_cache->Find(key); in CompileImpl()309 JitExecutableCache::Entry entry = jit_executable_cache->Allocate(key); in CompileImpl()554 JitExecutableCache* jit_executable_cache = state->jit_executable_cache; in WaitForCompilation() local555 if (auto cached = jit_executable_cache->Find(kernel.id())) in WaitForCompilation()
24 TfJitRtRequestState::JitExecutableCache* jit_executable_cache = in SetUpTfJitRtRequestContext() local29 builder->context_data().emplace<TfJitRtRequestState>(jit_executable_cache); in SetUpTfJitRtRequestContext()
129 JitExecutableCache* jit_executable_cache = state->jit_executable_cache; in CompileImpl() local140 if (auto cached = jit_executable_cache->Find(key)) return cached; in CompileImpl()150 JitExecutableCache::Entry entry = jit_executable_cache->Allocate(key); in CompileImpl()