Home
last modified time | relevance | path

Searched refs:is_var_sigmoid2 (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtf_lstm_cell_fusion.cc84 auto is_var_sigmoid2 = std::make_shared<Var>("Sigmoid"); in GenerateBodyGraphHiddenPattern() local
85 MS_CHECK_TRUE_RET(is_var_sigmoid2 != nullptr, {}); in GenerateBodyGraphHiddenPattern()
86 VectorRef forget_gate = VectorRef({is_var_sigmoid2, forget_bias}); in GenerateBodyGraphHiddenPattern()
Dtflite_lstm_cell_fusion.cc104 auto is_var_sigmoid2 = std::make_shared<Var>("Sigmoid"); in GenerateBodyGraphCellPattern() local
105 MS_CHECK_TRUE_RET(is_var_sigmoid2 != nullptr, {}); in GenerateBodyGraphCellPattern()
106 VectorRef forget_gate = VectorRef({is_var_sigmoid2, bias_forget}); in GenerateBodyGraphCellPattern()