Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/mps/
DOperationUtils.h208 …CacheEntry(const std::string& key, MPSCachedGraph *cachedGraph) : cachedGraph_(cachedGraph), key_(… in CacheEntry()
209 MPSCachedGraph* cachedGraph_ = nullptr; member
226 delete i.second.cachedGraph_; in ~MPSGraphCache()
245 cachedGraph = entry.cachedGraph_; in CreateCachedGraph()
272 cachedGraph = entry.cachedGraph_; in LookUp()
DOperationUtils.mm746 profiler.beginProfileKernel(cacheEntry.cachedGraph_->graph(), graphKey, true);