Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtflite_lstm_cell_fusion.h82 mutable std::vector<VarPtr> while_input_vars_; variable
Dtflite_lstm_cell_fusion.cc154 while_input_vars_.emplace_back(is_var); in Init()
291 auto while_inputs = while_input_vars_; in DefinePattern()
293 while_inputs.insert(while_inputs.begin() + 4, while_input_vars_[2]); in DefinePattern()
531 auto &vars = while_input_vars_; in CreateLSTMNode()
Dtf_lstm_cell_fusion.cc366 auto &vars = while_input_vars_; in CreateLSTMNode()