Searched refs:is_make_tuple (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tf_lstm_cell_fusion.cc | 170 auto is_make_tuple = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimMakeTuple)); in GetBodyGraphPattern() local
171 MS_CHECK_TRUE_RET(is_make_tuple != nullptr, nullptr); in GetBodyGraphPattern()
172 …std::vector<BaseRef> outputs = {is_make_tuple, add3, placeholders[1], add2, set_item, output_cell,… in GetBodyGraphPattern()
|
| D | tflite_lstm_cell_fusion.cc | 268 auto is_make_tuple = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimMakeTuple)); in GetBodyGraphPattern() local
269 MS_CHECK_TRUE_RET(is_make_tuple != nullptr, nullptr); in GetBodyGraphPattern()
270 …std::vector<BaseRef> outputs = {is_make_tuple, add3, placeholders[1], add2, set_item, cell_output,… in GetBodyGraphPattern()
|
| D | tf_bidirection_gru_fusion.cc | 361 auto is_make_tuple = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimMakeTuple)); in GetBodyGraphPattern() local
362 MS_CHECK_TRUE_RET(is_make_tuple != nullptr, nullptr); in GetBodyGraphPattern()
363 std::vector<BaseRef> outputs = {is_make_tuple, add1, placeholders[1], add, in GetBodyGraphPattern()
|