Searched defs:key_node (Results 1 – 11 of 11) sorted by relevance
195 for (const auto &key_node : backward_matmul_dx_dw_map) { in ExtractDxGradCommMap() local
714 auto key_node = NewValueNode(std::make_shared<StringImm>(internal_arg)); in ConvertCNodeToPyInterpretForPrim() local743 auto key_node = NewValueNode(std::make_shared<StringImm>(internal_arg)); in ConvertCNodeToPyExecuteForPrim() local
312 expr_ty key_node; in append_ast_dict() local
438 …auto key_node = fa_score_node->input(ops::FlashAttentionScoreInputIndex::kFlashAttentionScoreInput… in CreateReplaceFSPGraph() local
560 auto key_node = NewValueNode(elements[i].first->BuildValue()); in EraseDictItems() local1776 const auto key_node = NewValueNode(kConvertToListKey); in ConvertJoinedStr() local
691 ValueNode *key_node = call_node->input(1 + is_method_descriptor); in InferDictPop() local
3666 auto key_node = fg_builder_->AddNode(prim::kPrimMakeTuple, key_obj); in HandleBuildOp() local
273 auto key_node = cur_node->cast<CNodePtr>()->input(key_inputs_index); in ConvertToPyInterpretCall() local
379 …AnfNodePtr key_node = MixedPrecisionCastHelper(dict_key_node, item.first, target_type, func_graph); in MixedPrecisionCastHelper() local
2641 AnfNodePtr key_node = ParseExprNode(block, keys[index]); in GetRealKeysValues() local3650 AnfNodePtr key_node = ParseExprNode(dict_body_block, key_obj); in ParseDictCompIfs() local
855 const AnfNodePtr &key_node) { in AddVisitedNode()