Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dlazy_graph_executor.cpp540 ComputationCache::TypePtr cached_computation) in Async() argument
546 cached_computation(std::move(cached_computation)), in Async()
743 ComputationCache::TypePtr cached_computation = in TryRunCachedSync() local
745 if (cached_computation == nullptr) { in TryRunCachedSync()
749 auto* comp = cached_computation->computation.get(); in TryRunCachedSync()
759 std::move(cached_computation), in TryRunCachedSync()
813 ComputationCache::TypePtr cached_computation = in LookupCachedCompile() local
815 if (cached_computation == nullptr) { in LookupCachedCompile()
820 return cached_computation; in LookupCachedCompile()
867 auto cached_computation = std::make_shared<CachedComputation>( in SyncTensorsGraphInternal() local
[all …]
H A Dlazy_graph_executor.h319 ComputationCache::TypePtr cached_computation);
329 ComputationCache::TypePtr cached_computation; member
401 ComputationCache::TypePtr cached_computation);
407 ComputationCache::TypePtr cached_computation,