Searched defs:k_input (Results 1 – 4 of 4) sorted by relevance
29 const TensorC *k_input = inputs[SECOND_INPUT]; in AttentionInferShape() local
58 auto k_input = cnode->input(kInputNumThree - 1); in Mapper() local
268 auto k_input = std::make_shared<Var>(); // input K in DefineFlashAttentionPattern() local382 auto k_input = std::make_shared<Var>(); // input K in DefineFlashAttentionPattern() local
260 bool IpAdapterPattern(const CNodePtr q_input, const CNodePtr k_input) { in IpAdapterPattern()459 auto k_input = std::make_shared<Var>(); in DefineFlashAttentionPatternForMsSD21() local880 auto k_input = std::make_shared<Var>(); // input Q in DefineFlashAttentionPatternForSDPreMul() local1841 auto k_input = k_reshape->input(kNumIndex1)->cast<CNodePtr>(); in CreateFlashAttentionNodeForMsSDPseShift() local