Searched refs:while_cnode (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | control_flow_pass.cc | 223 …const FuncGraphPtr &fg, const CNodePtr &while_cnode, const std::vector<AnfNodePtr> &visited_nodes_… in CreateWhileCondCallNode() argument
226 auto cond_vnode = while_cnode->input(kWhileCondIndex); in CreateWhileCondCallNode()
242 …cond_partial_cnode_inputs.insert(cond_partial_cnode_inputs.end(), while_cnode->inputs().begin() + … in CreateWhileCondCallNode()
243 while_cnode->inputs().end()); in CreateWhileCondCallNode()
286 …ntrolFlowPass::CreateWhileBodyPartialNode(const FuncGraphPtr &cond_fg, const CNodePtr &while_cnode, in CreateWhileBodyPartialNode() argument
288 auto body_vnode = while_cnode->input(kWhileBodyIndex); in CreateWhileBodyPartialNode()
358 CNodePtr *while_cnode, CNodePtr *after_partial_cnode) { in CreateWhileAfterPartialNode() argument
361 if (CreateAfterGraph(main_fg, remain_nodes, *while_cnode, &after_fg) != RET_SUCCESS) { in CreateWhileAfterPartialNode()
384 if (get_tuple_item_cnode->input(kCNodeFirstInputIndex) != *while_cnode) { in CreateWhileAfterPartialNode()
433 auto while_cnode = while_node->cast<CNodePtr>(); in ProcessWhileOp() local
[all …]
|
| D | control_flow_pass.h | 51 …const FuncGraphPtr &fg, const CNodePtr &while_cnode, const std::vector<AnfNodePtr> &visited_nodes_…
54 …int CreateWhileBodyPartialNode(const FuncGraphPtr &cond_fg, const CNodePtr &while_cnode, CNodePtr …
59 CNodePtr *while_cnode, CNodePtr *after_partial_cnode);
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tflite_lstm_cell_fusion.cc | 326 …tmCellFusion::CheckReferencedOutputs(const FuncGraphPtr &func_graph, const CNodePtr &while_cnode) { in CheckReferencedOutputs() argument
328 MS_ASSERT(while_cnode != nullptr); in CheckReferencedOutputs()
334 auto while_node_users = manager->node_users()[while_cnode]; in CheckReferencedOutputs()
376 MS_ASSERT(while_cnode != nullptr); in CheckBodyGraph()
630 …iteLstmCellFusion::AdjustOtherGetItems(const FuncGraphPtr &func_graph, const CNodePtr &while_cnode, in AdjustOtherGetItems() argument
632 MS_ASSERT(func_graph != nullptr && while_cnode != nullptr); in AdjustOtherGetItems()
639 auto while_node_users = manager->node_users()[while_cnode]; in AdjustOtherGetItems()
750 auto while_cnode = utils::cast<CNodePtr>(while_node); in Process() local
752 if (while_cnode == nullptr || while_cnode->size() != while_inputs_num_) { in Process()
755 if (!CheckReferencedOutputs(func_graph, while_cnode)) { in Process()
[all …]
|
| D | tflite_lstm_cell_fusion.h | 58 …tatic lite::STATUS AdjustOtherGetItems(const FuncGraphPtr &func_graph, const CNodePtr &while_cnode,
72 static bool CheckReferencedOutputs(const FuncGraphPtr &func_graph, const CNodePtr &while_cnode);
|