Home
last modified time | relevance | path

Searched defs:UpdateCache (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dupdate_cache.cc51 class MIND_API UpdateCache : public BaseOperator { class
55 UpdateCache() : BaseOperator("UpdateCache") {} in UpdateCache() function in mindspore::ops::UpdateCache
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
D_embedding_cache_ops.py24 class UpdateCache(PrimitiveWithCheck): class
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/kernel_builder/ascend/ascendc/op_kernel/
Dprompt_kv_cache.cpp52 __aicore__ inline void UpdateCache(GM_ADDR cache, GM_ADDR update) { in UpdateCache() function in KernelPromptKvCache
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/ascendc/op_kernel/
Dprompt_kv_cache.cpp52 __aicore__ inline void UpdateCache(GM_ADDR cache, GM_ADDR update) { in UpdateCache() function in KernelPromptKvCache