Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc1578 AnfNodePtr *const first_cache_input_index, in GetFirstPSEmbeddingCache() argument
1617 *first_cache_input_index = cnode; in GetFirstPSEmbeddingCache()
1650 AnfNodePtr first_cache_input_index = nullptr; in CheckIfSupportPSEmbeddingCache() local
1652 GetFirstPSEmbeddingCache(graph, &first_cache_input_index, &first_cache_size); in CheckIfSupportPSEmbeddingCache()
1653 MS_EXCEPTION_IF_NULL(first_cache_input_index); in CheckIfSupportPSEmbeddingCache()
1678 if (cnode == first_cache_input_index) { in CheckIfSupportPSEmbeddingCache()
Dkernel_runtime.h171 …FirstPSEmbeddingCache(const session::KernelGraph &graph, AnfNodePtr *const first_cache_input_index,