Searched refs:is_param1 (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | multi_head_attention_fusion.cc | 48 auto is_param1 = std::make_shared<CondVar>(IsParamNode); in DefineMask() local
49 MS_CHECK_TRUE_RET(is_param1 != nullptr, {}); in DefineMask()
50 auto expand_dims = VectorRef({is_expand_dims, mask_input, is_param1}); in DefineMask()
76 auto is_param1 = std::make_shared<CondVar>(IsParamNode); in DefineMPWithMaskPattern() local
77 MS_CHECK_TRUE_RET(is_param1 != nullptr, {}); in DefineMPWithMaskPattern()
78 auto q2k_normed = VectorRef({is_mul, q2k, is_param1}); in DefineMPWithMaskPattern()
109 auto is_param1 = std::make_shared<CondVar>(IsParamNode); in DefineDensePattern() local
110 MS_CHECK_TRUE_RET(is_param1 != nullptr, {}); in DefineDensePattern()
111 auto transpose = VectorRef({is_tranpose, input, is_param1}); in DefineDensePattern()
153 auto is_param1 = std::make_shared<CondVar>(IsParamNode); in DefineProcessOutputPattern() local
[all …]
|
| D | squeeze_fusion.cc | 31 auto is_param1 = std::make_shared<CondVar>(IsParamNode); in DefinePattern() local
32 MS_CHECK_TRUE_RET(is_param1 != nullptr, {}); in DefinePattern()
37 VectorRef bn_ref = VectorRef({is_bn, is_squeeze, is_param1, is_param2, is_seq_var}); in DefinePattern()
|
| D | tf_bidirection_gru_fusion.cc | 129 auto is_param1 = std::make_shared<CondVar>(IsParameterNode); in DefineFowardPattern() local
130 MS_CHECK_TRUE_RET(is_param1 != nullptr, {}); in DefineFowardPattern()
131 auto fw_reduce = VectorRef({is_reduce, input_length_, is_param1}); in DefineFowardPattern()
195 auto is_param1 = std::make_shared<CondVar>(IsParameterNode); in DefinebackwardPattern() local
196 MS_CHECK_TRUE_RET(is_param1 != nullptr, {}); in DefinebackwardPattern()
197 auto bw_max1 = VectorRef({is_reduce, input_length_, is_param1}); in DefinebackwardPattern()
282 auto is_param1 = std::make_shared<CondVar>(IsParameterNode); in GetCondGraphPattern() local
283 MS_CHECK_TRUE_RET(is_param1 != nullptr, nullptr); in GetCondGraphPattern()
286 VectorRef less1_ref = VectorRef({is_less1, is_param1, is_param2}); in GetCondGraphPattern()
315 auto is_param1 = std::make_shared<CondVar>(IsParameterNode); in GetBodyGraphPattern() local
[all …]
|
| D | tf_lstm_cell_fusion.cc | 128 auto is_param1 = std::make_shared<CondVar>(IsParameterNode); in GetBodyGraphPattern() local
129 MS_CHECK_TRUE_RET(is_param1 != nullptr, nullptr); in GetBodyGraphPattern()
130 VectorRef add2 = VectorRef({is_var1, placeholders[2], is_param1}); in GetBodyGraphPattern()
|
| D | tflite_rel_pos_multi_head_attention_fusion.cc | 88 auto is_param1 = std::make_shared<CondVar>(IsParamNode); in DefinePatterns() local
89 MS_CHECK_TRUE_RET(is_param1 != nullptr, {}); in DefinePatterns()
90 query_with_bias_u = VectorRef({is_transpose1, query_with_bias_u, is_param1}); in DefinePatterns()
|
| D | tflite_lstm_cell_fusion.cc | 219 auto is_param1 = std::make_shared<CondVar>(IsParameterNode); in GetBodyGraphPattern() local
220 MS_CHECK_TRUE_RET(is_param1 != nullptr, nullptr); in GetBodyGraphPattern()
221 VectorRef add2 = VectorRef({is_var1, placeholders[2], is_param1}); in GetBodyGraphPattern()
|