| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | conv_activation_fusion.cc | 54 AnfNodePtr pre_node = act_node->input(1); in Process() local
55 if (pre_node != nullptr && pre_node->isa<CNode>()) { in Process()
56 if (IsMultiOutputTensors(func_graph, pre_node)) { in Process()
59 auto conv_node = pre_node->cast<CNodePtr>(); in Process()
72 return pre_node; in Process()
|
| D | affine_activation_fusion.cc | 55 AnfNodePtr pre_node = activation_node->input(1); in Process() local
56 if (!CheckPrimitiveType(pre_node, prim::kPrimAffine)) { in Process()
61 auto affine_node = pre_node->cast<CNodePtr>(); in Process()
|
| D | squeeze_fusion.cc | 84 auto pre_node = squeeze_cnode->input(1); in Process() local
102 (void)manager->Replace(squeeze_node, pre_node); in Process()
103 return pre_node; in Process()
|
| D | affine_fusion.cc | 64 AnfNodePtr pre_node = matmul_node->input(1); in Process() local
65 if (!CheckPrimitiveType(pre_node, prim::kPrimSplice)) { in Process()
70 auto splice_node = pre_node->cast<CNodePtr>(); in Process()
|
| D | conv_transform_fusion.cc | 103 auto pre_node = transform_node->input(1); in Process() local
104 auto conv_node = pre_node->cast<CNodePtr>(); in Process()
142 pre_node->set_abstract(abstr); in Process()
143 return pre_node; in Process()
|
| D | transpose_fusion.cc | 158 auto pre_node = trans_cnode_2->input(1); in TransTransFusion() local
159 auto pre_cnode = pre_node->cast<CNodePtr>(); in TransTransFusion()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_build_ascend.cc | 131 static std::vector<size_t> CalCleanZerosSize(const CNodePtr &pre_node) { in CalCleanZerosSize() argument
132 MS_EXCEPTION_IF_NULL(pre_node); in CalCleanZerosSize()
133 auto kernel_mod = AnfAlgo::GetKernelMod(pre_node); in CalCleanZerosSize()
138 if (AnfAlgo::HasNodeAttr(kAttrAtomicOutputIndexs, pre_node)) { in CalCleanZerosSize()
139 … auto output_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicOutputIndexs); in CalCleanZerosSize()
147 if (AnfAlgo::HasNodeAttr(kAttrAtomicWorkspaceIndexs, pre_node)) { in CalCleanZerosSize()
148 …auto workspace_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicWorkspaceIn… in CalCleanZerosSize()
155 …MS_LOG(INFO) << "clear output size:" << clean_size_list.size() << ",pre_node:" << pre_node->fullna… in CalCleanZerosSize()
160 … const mindspore::CNodePtr &pre_node, std::vector<mindspore::CNodePtr> *new_nodes) { in AddTbeClearZeroNode() argument
162 MS_EXCEPTION_IF_NULL(pre_node); in AddTbeClearZeroNode()
[all …]
|
| D | ascend_stream_assign.cc | 1044 auto pre_node = AnfAlgo::GetCNodeName(cnode_ptr_list[i - 1]); in InsertStreamActiveForCommon() local
1045 if (pre_node == kLabelSwitchOpName || pre_node == kLabelGotoOpName) { in InsertStreamActiveForCommon()
|
| /third_party/musl/porting/linux/user/ldso/ |
| D | dynlink_rand.c | 80 struct handle_node *pre_node = NULL; in remove_handle_node() local
83 if (pre_node) { in remove_handle_node()
84 pre_node->next = node->next; in remove_handle_node()
91 pre_node = node; in remove_handle_node()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.h | 61 const FuncGraphPtr &func_graph, int64_t pos, const CNodePtr &pre_node);
63 TensorLayout GetTensorInLayout(const CNodePtr &pre_node, const PrimitivePtr &pre_prim,
70 const CNodePtr &pre_node);
79 … const TensorRedistribution &tensor_redistribution, const CNodePtr &pre_node);
|
| D | step_parallel.cc | 197 void InsertNode(const Operator &op, const CNodePtr &node, size_t index, const AnfNodePtr &pre_node, in InsertNode() argument
207 node_input = CreateMirrorInput(root, op, pre_node, instance_name, param_name); in InsertNode()
209 node_input = CreateInput(op, pre_node, instance_name); in InsertNode()
231 static CNodePtr ReplaceNode(const Operator &op, const AnfNodePtr &pre_node, const FuncGraphPtr &fun… in ReplaceNode() argument
237 ScopePtr scope = pre_node->scope(); in ReplaceNode()
241 node_input = CreateMirrorInput(root, op, pre_node, instance_name, param_name); in ReplaceNode()
243 node_input = CreateInput(op, pre_node, instance_name); in ReplaceNode()
258 manager->Replace(pre_node, new_node); in ReplaceNode()
328 const FuncGraphPtr &func_graph, int64_t pos, const CNodePtr &pre_node) { in InsertRedistribution() argument
330 MS_EXCEPTION_IF_NULL(pre_node); in InsertRedistribution()
[all …]
|
| D | step_auto_parallel.cc | 860 auto pre_node = cnode->input(1); in ReshapeCostCompute() local
861 if (IsPrimitiveCNode(pre_node, prim::kPrimLoad)) { in ReshapeCostCompute()
862 pre_node = pre_node->cast<CNodePtr>()->input(1); in ReshapeCostCompute()
868 if (pre_node->isa<Parameter>()) { in ReshapeCostCompute()
874 if (!FindReshapePreNodeStraCosts(pre_node, &pre_operator_info, &out_index, 0)) { in ReshapeCostCompute()
898 bool is_prev_param = pre_node->isa<Parameter>(); in ReshapeCostCompute()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | redundant_op_remove_pass.cc | 78 AnfNodePtr pre_node = cnode->input(1); in ProcessDependencyWithTwoNodes() local
80 MS_ASSERT(pre_node != nullptr); in ProcessDependencyWithTwoNodes()
83 pre_node = cnode->input(kInputIndexTwo); in ProcessDependencyWithTwoNodes()
86 if (CheckPrimitiveType(pre_node, prim::kPrimTranspose)) { in ProcessDependencyWithTwoNodes()
87 pre_node = cnode->input(1)->cast<CNodePtr>()->input(1); in ProcessDependencyWithTwoNodes()
88 MS_CHECK_TRUE_MSG(pre_node != nullptr, RET_ERROR, "pre_node is nullptr"); in ProcessDependencyWithTwoNodes()
96 auto node_users = manager->node_users()[pre_node]; in ProcessDependencyWithTwoNodes()
107 auto depend_node = func_graph->NewCNode(depend_prim, {post_node, pre_node}); in ProcessDependencyWithTwoNodes()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 1189 auto pre_node = (cnode->inputs()[1])->cast<CNodePtr>(); in GenAddrCleanLaunchArgs() local
1191 if (AnfAlgo::HasNodeAttr(kAttrAtomicOutputIndexs, pre_node)) { in GenAddrCleanLaunchArgs()
1193 …auto clean_output_indexes = AnfAlgo::GetNodeAttr<std::vector<int>>(pre_node, kAttrAtomicOutputInde… in GenAddrCleanLaunchArgs()
1195 …auto clean_output_indexes = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicOutputI… in GenAddrCleanLaunchArgs()
1198 auto device_address = AnfAlgo::GetOutputAddr(pre_node, index); in GenAddrCleanLaunchArgs()
1213 if (AnfAlgo::HasNodeAttr(kAttrAtomicWorkspaceIndexs, pre_node)) { in GenAddrCleanLaunchArgs()
1215 …auto clean_workspaces_indexes = AnfAlgo::GetNodeAttr<std::vector<int>>(pre_node, kAttrAtomicWorksp… in GenAddrCleanLaunchArgs()
1217 …auto clean_workspaces_indexes = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicWor… in GenAddrCleanLaunchArgs()
1220 auto device_address = AnfAlgo::GetWorkspaceAddr(pre_node, index); in GenAddrCleanLaunchArgs()
1627 auto pre_node = AnfAlgo::GetPrevNodeOutput(node, 1, true); in CheckSparsePSEmbeddingCache() local
[all …]
|
| D | kernel_adjust.cc | 185 auto pre_node = orders[idx - 1]; in CopyMemcpyList() local
186 auto pre_kernel_name = AnfAlgo::GetCNodeName(pre_node); in CopyMemcpyList()
189 (*memcpy_list).push_back(pre_node); in CopyMemcpyList()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/ |
| D | task_generator.cc | 108 auto pre_node = anf_node_ptr->input(i + 1)->cast<CNodePtr>(); in LaunchAddrCleanKernel() local
109 if (AnfAlgo::HasNodeAttr(kAttrAtomicOutputIndexs, pre_node)) { in LaunchAddrCleanKernel()
110 …auto clean_output_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicOutputIn… in LaunchAddrCleanKernel()
112 auto device_address = AnfAlgo::GetOutputAddr(pre_node, index); in LaunchAddrCleanKernel()
123 if (AnfAlgo::HasNodeAttr(kAttrAtomicWorkspaceIndexs, pre_node)) { in LaunchAddrCleanKernel()
124 …auto clean_workspace_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicWorks… in LaunchAddrCleanKernel()
126 auto device_address = AnfAlgo::GetWorkspaceAddr(pre_node, index); in LaunchAddrCleanKernel()
|
| /third_party/mindspore/mindspore/lite/tools/converter/acl/ |
| D | acl_pass.cc | 266 AnfNodePtr pre_node = nullptr; in TraceOutput() local
268 pre_node = cur_node; in TraceOutput()
298 if (pre_node != nullptr && IsPrimitiveCNode(pre_node, prim::kPrimTupleGetItem)) { in TraceOutput()
299 cnode = pre_node->cast<CNodePtr>(); in TraceOutput()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.cc | 57 auto pre_node = cnode->input(1); in FindAccuGrad() local
64 if (pre_node->isa<Parameter>()) { in FindAccuGrad()
65 return pre_node; in FindAccuGrad()
67 if (pre_node->isa<CNode>()) { in FindAccuGrad()
68 auto pre_cnode = pre_node->cast<CNodePtr>(); in FindAccuGrad()
69 pre_node = pre_cnode->input(1); in FindAccuGrad()
|
| D | node_info.cc | 333 auto pre_node = cnode->input(1); in FindReshapePreNodeStraCosts() local
334 if (!pre_node->isa<CNode>()) { in FindReshapePreNodeStraCosts()
337 CNodePtr pre_cnode = pre_node->cast<CNodePtr>(); in FindReshapePreNodeStraCosts()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | parallel_fusion.cc | 62 for (const auto &pre_node : predecessors) { in ProcessThroughPassCNode() local
63 (*node_rels)[cur_node].pres.erase(pre_node); in ProcessThroughPassCNode()
64 (*node_rels)[pre_node].nexts.erase(cur_node); in ProcessThroughPassCNode()
65 node_que.push(pre_node); in ProcessThroughPassCNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/ |
| D | somas.cc | 589 auto pre_node = kernel->input(i + 1)->cast<CNodePtr>(); in InitAtomicCleanInputs() local
590 auto iter = nodes_map_.find(pre_node.get()); in InitAtomicCleanInputs()
593 << pre_node->fullname_with_scope() << "] is not init."; in InitAtomicCleanInputs()
598 if (AnfAlgo::HasNodeAttr(kAttrAtomicOutputIndexs, pre_node)) { in InitAtomicCleanInputs()
599 …auto clean_output_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicOutputIn… in InitAtomicCleanInputs()
602 …MS_LOG(EXCEPTION) << "Output index " << index << " exceed input node [" << pre_node->fullname_with… in InitAtomicCleanInputs()
616 if (AnfAlgo::HasNodeAttr(kAttrAtomicWorkspaceIndexs, pre_node)) { in InitAtomicCleanInputs()
617 …auto clean_workspace_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicWorks… in InitAtomicCleanInputs()
620 …MS_LOG(EXCEPTION) << "Workspace index " << index << " exceed input node [" << pre_node->fullname_w… in InitAtomicCleanInputs()
|
| /third_party/mindspore/tests/ut/cpp/parallel/ |
| D | step_parallel_test.cc | 430 CNodePtr pre_node = inputs[1]->cast<CNodePtr>(); in TEST_F() local
431 … PrimitivePtr pre_prim = pre_node->input(0)->cast<ValueNodePtr>()->value()->cast<PrimitivePtr>(); in TEST_F()
432 CNodePtr pre_node2 = pre_node->input(1)->cast<CNodePtr>(); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 2444 auto pre_node = (cnode->inputs()[1])->cast<CNodePtr>(); in CacheAddrForAtomicClean() local
2446 if (HasNodeAttr(kAttrAtomicOutputIndexs, pre_node)) { in CacheAddrForAtomicClean()
2448 auto clean_output_indexes = GetNodeAttr<std::vector<int>>(pre_node, kAttrAtomicOutputIndexs); in CacheAddrForAtomicClean()
2450 auto clean_output_indexes = GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicOutputIndexs); in CacheAddrForAtomicClean()
2453 auto device_address = GetOutputAddr(pre_node, index); in CacheAddrForAtomicClean()
2464 if (HasNodeAttr(kAttrAtomicWorkspaceIndexs, pre_node)) { in CacheAddrForAtomicClean()
2466 …auto clean_workspaces_indexes = GetNodeAttr<std::vector<int>>(pre_node, kAttrAtomicWorkspaceIndexs… in CacheAddrForAtomicClean()
2468 …auto clean_workspaces_indexes = GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicWorkspaceInd… in CacheAddrForAtomicClean()
2471 auto device_address = GetWorkspaceAddr(pre_node, index); in CacheAddrForAtomicClean()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | convert.cc | 625 AnfNodePtr pre_node = nullptr; in TraceOutput() local
646 pre_node = anf_out; in TraceOutput()
672 if ((pre_node != nullptr) && IsPrimitiveCNode(pre_node, prim::kPrimTupleGetItem)) { in TraceOutput()
673 auto item = out_handle_cache_.find(pre_node.get()); in TraceOutput()
|