Home
last modified time | relevance | path

Searched refs:is_var_mul3 (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtf_lstm_cell_fusion.cc156 auto is_var_mul3 = std::make_shared<Var>("Mul"); in GetBodyGraphPattern() local
157 MS_CHECK_TRUE_RET(is_var_mul3 != nullptr, {}); in GetBodyGraphPattern()
158 VectorRef new_to_hidden = VectorRef({is_var_mul3, hidden_zoneout_new_, new_hidden}); in GetBodyGraphPattern()
Dtflite_lstm_cell_fusion.cc250 auto is_var_mul3 = std::make_shared<Var>("Mul"); in GetBodyGraphPattern() local
251 MS_CHECK_TRUE_RET(is_var_mul3 != nullptr, {}); in GetBodyGraphPattern()
252 VectorRef output = VectorRef({is_var_mul3, output_gate, cell_to_output}); in GetBodyGraphPattern()
Dtf_bidirection_gru_fusion.cc83 auto is_var_mul3 = std::make_shared<Var>("Mul"); in GenerateBodyGraphHiddenPattern() local
84 MS_CHECK_TRUE_RET(is_var_mul3 != nullptr, {}); in GenerateBodyGraphHiddenPattern()
85 VectorRef updated = VectorRef({is_var_mul3, minus_update, tanh}); in GenerateBodyGraphHiddenPattern()