Searched refs:is_tensor_list_stack (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tf_bidirection_gru_fusion.cc | 176 …auto is_tensor_list_stack = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTensorLis… in DefineFowardPattern() local
177 MS_CHECK_TRUE_RET(is_tensor_list_stack != nullptr, {}); in DefineFowardPattern()
180 auto fw_stack = VectorRef({is_tensor_list_stack, fw_get_item, is_param7}); in DefineFowardPattern()
246 …auto is_tensor_list_stack = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTensorLis… in DefinebackwardPattern() local
247 MS_CHECK_TRUE_RET(is_tensor_list_stack != nullptr, {}); in DefinebackwardPattern()
250 auto bw_stack = VectorRef({is_tensor_list_stack, bw_get_item, is_param7}); in DefinebackwardPattern()
|
| D | tflite_lstm_cell_fusion.cc | 302 …auto is_tensor_list_stack = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTensorLis… in DefinePattern() local
303 MS_CHECK_TRUE_RET(is_tensor_list_stack != nullptr, {}); in DefinePattern()
306 VectorRef tensor_list_stack_node = VectorRef({is_tensor_list_stack, while_output, is_parameter}); in DefinePattern()
|