Searched refs:kPrimWhile (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/format/ |
| D | delete_redundant_transpose.cc | 35 if (CheckPrimitiveType(cnode, prim::kPrimIf) || CheckPrimitiveType(cnode, prim::kPrimWhile)) { in DeleteNot4DTranspose()
96 if (CheckPrimitiveType(cnode, prim::kPrimIf) || CheckPrimitiveType(cnode, prim::kPrimWhile)) { in TransTransFusion()
|
| D | to_format_base.cc | 301 …(opt::CheckPrimitiveType(node, prim::kPrimIf) || opt::CheckPrimitiveType(node, prim::kPrimWhile)) { in BasicProcess()
347 if (CheckPrimitiveType(node, prim::kPrimIf) || CheckPrimitiveType(node, prim::kPrimWhile)) { in ConvWeightFormatTrans()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | infershape_pass.cc | 140 if (CheckPrimitiveType(node, prim::kPrimIf) || CheckPrimitiveType(node, prim::kPrimWhile)) { in JudgeAllOpsCanInfer()
181 …(opt::CheckPrimitiveType(node, prim::kPrimIf) || opt::CheckPrimitiveType(node, prim::kPrimWhile)) { in InferProcess()
|
| D | decrease_transpose_algo.cc | 663 if (CheckPrimitiveType(node, prim::kPrimIf) || CheckPrimitiveType(node, prim::kPrimWhile)) { in DecreaseTransposeForSingleOp()
743 if (CheckPrimitiveType(node, prim::kPrimIf) || CheckPrimitiveType(node, prim::kPrimWhile)) { in DecreaseTransposeForMultiOp()
798 if (CheckPrimitiveType(cnode, prim::kPrimIf) || CheckPrimitiveType(cnode, prim::kPrimWhile)) { in DoFixFormat()
|
| D | control_flow_pass.cc | 154 (CheckPrimitiveType(node, prim::kPrimWhile) || CheckPrimitiveType(node, prim::kPrimIf))) { in SplitGraph()
764 if (CheckPrimitiveType(control_flow_node, prim::kPrimWhile)) { in ProcessControlOp()
|
| D | redundant_op_remove_pass.cc | 392 if (CheckPrimitiveType(node, prim::kPrimIf) || CheckPrimitiveType(node, prim::kPrimWhile)) { in Run()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | unify_format.cc | 348 …(opt::CheckPrimitiveType(node, prim::kPrimIf) || opt::CheckPrimitiveType(node, prim::kPrimWhile)) { in ProcessResizeAndFormat()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tf_bidirection_gru_fusion.cc | 159 auto is_while = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimWhile)); in DefineFowardPattern()
229 auto is_while = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimWhile)); in DefinebackwardPattern()
|
| D | tflite_lstm_cell_fusion.cc | 288 auto is_while_node = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimWhile)); in DefinePattern()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_model_parser.cc | 617 if (opt::CheckPrimitiveType(cnode, prim::kPrimWhile)) { in ConvertSubgraphInputs()
694 if (opt::CheckPrimitiveType(cnode, prim::kPrimWhile)) { in UpdateMap()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_model_parser.cc | 553 if (opt::CheckPrimitiveType(control_flow_node, prim::kPrimWhile)) { in BuildSubFuncGraphMap()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 594 inline const PrimitivePtr kPrimWhile = std::make_shared<Primitive>("While"); variable
|