Searched refs:GetFirstPSEmbeddingCache (Results 1 – 2 of 2) sorted by relevance
171 …void GetFirstPSEmbeddingCache(const session::KernelGraph &graph, AnfNodePtr *const first_cache_inp…
1577 void KernelRuntime::GetFirstPSEmbeddingCache(const session::KernelGraph &graph, in GetFirstPSEmbeddingCache() function in mindspore::device::KernelRuntime1652 GetFirstPSEmbeddingCache(graph, &first_cache_input_index, &first_cache_size); in CheckIfSupportPSEmbeddingCache()