Searched refs:SynchronizeStream (Results 1 – 6 of 6) sorted by relevance
40 virtual bool SynchronizeStream() = 0;
412 RETURN_IF_FALSE(embedding_device_cache_->cache_->SynchronizeStream()); in ProcessData()823 RETURN_IF_FALSE(embedding_device_cache_->cache_->SynchronizeStream()); in HashSwapHostToDevice()851 RETURN_IF_FALSE(embedding_device_cache_->cache_->SynchronizeStream()); in HashSwapDeviceToHost()1101 RETURN_IF_FALSE(embedding_device_cache_->cache_->SynchronizeStream()); in SyncDeviceEmbeddingTable()1136 embedding_device_cache_->cache_->SynchronizeStream(); in DumpHashTables()
35 bool SynchronizeStream() override;
60 bool GPUPsCache::SynchronizeStream() { in SynchronizeStream() function in mindspore::ps::gpu::GPUPsCache
58 bool SynchronizeStream() override;
146 return SynchronizeStream(); in MallocConstantMemory()180 bool AscendPsCache::SynchronizeStream() { in SynchronizeStream() function in mindspore::ps::ascend::AscendPsCache