Searched refs:kPrimIf (Results 1 – 10 of 10) 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 …if (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 …if (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()
772 if (CheckPrimitiveType(control_flow_node, prim::kPrimIf)) { 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 …if (opt::CheckPrimitiveType(node, prim::kPrimIf) || opt::CheckPrimitiveType(node, prim::kPrimWhile… in ProcessResizeAndFormat()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_model_parser.cc | 559 } else if (opt::CheckPrimitiveType(control_flow_node, prim::kPrimIf)) { in BuildSubFuncGraphMap()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_model_parser.cc | 702 if (opt::CheckPrimitiveType(cnode, prim::kPrimIf)) { in UpdateMap()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 666 inline const PrimitivePtr kPrimIf = std::make_shared<Primitive>("If"); variable
|