Home
last modified time | relevance | path

Searched refs:kPrimMapCacheIdx (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/core/abstract/
Dprimitive_infer_map.cc122 {prim::kPrimMapCacheIdx, {InferImplMapCacheIdx, nullptr, true}}, in GetPrimitiveToEvalImplMap()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc306 PrimitivePtr map_cache_primitive = prim::kPrimMapCacheIdx; in CreateMapCacheIdx()
/third_party/mindspore/mindspore/core/base/
Dcore_ops.h202 inline const PrimitivePtr kPrimMapCacheIdx = std::make_shared<Primitive>("MapCacheIdx"); variable