Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dencoder_layer_fusion.cc192 auto assign_k = VectorRef({is_assign_k, k_past_, mul_k, var_assign_k}); in DefinePatternInitReset() local
388 auto assign_k = VectorRef({is_assign_k, k_past_, DefineBatchValidLength(tuple_k), var_assign_k}); in DefineDependKV() local