| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/actor/ |
| D | switch_actor.cc | 87 for (auto &partial_node : partial_nodes_) { in AppendOutputTensors() local
112 for (auto &partial_node : partial_nodes_) { in ModifySubgraphKernel() local
193 for (auto &partial_node : partial_nodes_) { in CompileArrowThroughSwitchCall() local
|
| D | switch_actor.h | 34 delete partial_node; in ~LiteSwitchOpActor() local
|
| D | exit_actor.h | 44 kernel::KernelExec *partial_node = nullptr; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/kernel/ |
| D | exit_subgraph_kernel.cc | 31 void ExitSubGraphKernel::SetPartial(kernel::KernelExec *partial_node) { (void)partials_.insert(part… in SetPartial()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | grad_partial_transform.h | 52 auto partial_node = unpack_graph_cnode->input(1); in operator() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/ |
| D | control_flow_scheduler.cc | 76 auto &partial_node = item.first; in SplitNonTailCallSubGraphs() local
247 auto partial_node = reinterpret_cast<kernel::PartialFusionKernel *>(node->kernel()); in RecordNonTailCallLinkInfo() local
296 …owScheduler::RecordSubgraphCaller(const size_t &subgraph_index, kernel::KernelExec *partial_node) { in RecordSubgraphCaller()
422 for (auto partial_node : item.second) { in GetSubGraphsWhichNeedBoundary() local
442 for (auto partial_node : item.second) { in GetSubGraphsWhichNeedBoundary() local
486 for (auto partial_node : item.second) { in BuildBoundaryForMultipleCalledGraph() local
720 for (auto partial_node : subgraphs_need_boundary_[old_subgrpah]) { in IsolateInputOfMultipleCalledGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/legacy_optimizer/graph/ |
| D | infershape_pass.cc | 379 int64_t PartialGraphIndex(const CNodeT *partial_node) { in PartialGraphIndex()
385 int InferShapePass::CopyPartialShapeToSubGraph(const CNodeT *partial_node, MetaGraphT *graph) { in CopyPartialShapeToSubGraph()
418 void InferShapePass::RestoreSubGraphInput(const CNodeT *partial_node, MetaGraphT *graph) { in RestoreSubGraphInput()
429 …ss::SetNonTailCallOutputShape(const std::unique_ptr<CNodeT> &call_node, const CNodeT *partial_node, in SetNonTailCallOutputShape()
451 const CNodeT *partial_node, MetaGraphT *graph) { in InferPartialNode()
532 for (auto &partial_node : all_partial_nodes) { in InferSwitchOrSwitchLayerNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/ |
| D | kernel_exec_util.cc | 616 for (auto partial_node : partial_nodes) { in GetCallInputPartialsCorrespondingOutputSubgraph() local
634 KernelExec *KernelExecUtil::GetPartialOutputCall(const KernelExec *partial_node) { in GetPartialOutputCall()
697 KernelExec *KernelExecUtil::GetPartialOutputCall(const KernelExec *partial_node) { return nullptr; } in GetPartialOutputCall()
|
| D | scheduler.cc | 771 int Scheduler::RestoreSubGraphInput(const lite::LiteGraph::Node *partial_node) { in RestoreSubGraphInput()
789 int Scheduler::CopyPartialShapeToSubGraph(const lite::LiteGraph::Node *partial_node) { in CopyPartialShapeToSubGraph()
1503 bool Scheduler::IsControlFlowPattern(const lite::LiteGraph::Node &partial_node) { in IsControlFlowPattern()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/import/ |
| D | mindir_control_flow_adjust.cc | 248 auto partial_node = make_tuple_op_intpus[i]->cast<CNodePtr>(); in MoveCallInputsToPartialFusionInputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ir/ |
| D | func_graph_cloner.cc | 516 auto partial_node = replicated_func_graph[graph]; in SetPartialEdges() local
571 auto partial_node = replicated_func_graph[graph]; in SetEdges() local
647 auto partial_node = func_graph_user->NewCNodeWeak(std::move(cnode_inputs)); in AddInputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/ |
| D | kernel_graph_mgr.cc | 124 bool ExistGraphCaller(const AnfNodePtr &partial_node) { in ExistGraphCaller()
1333 auto partial_node = graph->NewCNode(partial_inputs); in CreateSwitchInput() local
1417 auto partial_node = node->cast<CNodePtr>(); in CreateCallSwitchInputs() local
1529 auto partial_node = partial_idx->cast<CNodePtr>(); in CreateCallSwitchLayerInputs() local
1580 auto partial_node = attr_input->cast<CNodePtr>(); in CreateSwitchOrPartialNode() local
1880 auto partial_node = last_input_node->cast<CNodePtr>(); in SetReturnNode() local
2603 std::string KernelGraphMgr::AddPartialParametersMap(const AnfNodePtr &partial_node) { in AddPartialParametersMap()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/ |
| D | node_pass.cc | 84 auto partial_node = dyn_cast<CNode>(node); in ModifyOutputAndCallerToMap() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/ |
| D | utils.cc | 147 auto partial_node = cnode->input(0)->cast<CNodePtr>(); in IsWhileNode() local
|
| D | convert.cc | 1215 auto partial_node = cnode->input(0); in CacheWhileGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/static_analysis/ |
| D | program_specialize.cc | 1007 auto partial_node = partial_closure->node(); in ProcessNode() local
1325 auto partial_node = partial_closure->node(); in BuildSpecializedParameterCNode() local
|
| D | prim.cc | 3017 auto partial_node = cnode->input(op_index); in EvalPrim() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/load_mindir/ |
| D | load_model.cc | 2412 auto partial_node = anf_node->cast<CNodePtr>(); in BuildAbstractFunction() local
|