Searched refs:is_var_mul2 (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tflite_lstm_cell_fusion.cc | 107 auto is_var_mul2 = std::make_shared<Var>("Mul"); in GenerateBodyGraphCellPattern() local
108 MS_CHECK_TRUE_RET(is_var_mul2 != nullptr, {}); in GenerateBodyGraphCellPattern()
109 VectorRef cell_forgeted = VectorRef({is_var_mul2, forget_gate, placeholders[4]}); in GenerateBodyGraphCellPattern()
233 auto is_var_mul2 = std::make_shared<Var>("Mul"); in GetBodyGraphPattern() local
234 MS_CHECK_TRUE_RET(is_var_mul2 != nullptr, {}); in GetBodyGraphPattern()
237 VectorRef zoneout_cell_new = VectorRef({is_var_mul2, cell_zoneout_new_, cell_new}); in GetBodyGraphPattern()
|
| D | tf_lstm_cell_fusion.cc | 147 auto is_var_mul2 = std::make_shared<Var>("Mul"); in GetBodyGraphPattern() local
148 MS_CHECK_TRUE_RET(is_var_mul2 != nullptr, {}); in GetBodyGraphPattern()
149 VectorRef old_to_cell = VectorRef({is_var_mul2, cell_zoneout_old_, placeholders[4]}); in GetBodyGraphPattern()
|
| D | tf_bidirection_gru_fusion.cc | 75 auto is_var_mul2 = std::make_shared<Var>("Mul"); in GenerateBodyGraphHiddenPattern() local
76 MS_CHECK_TRUE_RET(is_var_mul2 != nullptr, {}); in GenerateBodyGraphHiddenPattern()
77 VectorRef update_hidden = VectorRef({is_var_mul2, get_item2, placeholders[4]}); in GenerateBodyGraphHiddenPattern()
|