Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/lazy/python/
H A Dinit.cpp287 LazyGraphExecutor::Get()->GetComputationCache()->Get(hash); in initLazyBindings()
313 ->GetComputationCache() in initLazyBindings()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dlazy_graph_executor.cpp805 LazyGraphExecutor::ComputationCache* LazyGraphExecutor::GetComputationCache() { in GetComputationCache() function in torch::lazy::LazyGraphExecutor
814 GetComputationCache()->Get(hash); in LookupCachedCompile()
869 GetComputationCache()->Add(coll.hash, cached_computation); in SyncTensorsGraphInternal()
H A Dlazy_graph_executor.h131 ComputationCache* GetComputationCache();