Searched refs:temp_node (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.cc | 183 auto temp_node = temp_user.first; in HandleReceiveParam() local
185 if (IsPrimitiveCNode(temp_node, prim::kPrimCast)) { in HandleReceiveParam()
186 temp_node = node_users_map[temp_node].begin()->first; in HandleReceiveParam()
188 if (IsPrimitiveCNode(temp_node, prim::kPrimMirrorMicroStep) || in HandleReceiveParam()
189 IsPrimitiveCNode(temp_node, prim::kPrimMicroStepAllGather)) { in HandleReceiveParam()
190 auto node_set = node_users_map[temp_node]; in HandleReceiveParam()
213 auto temp_node = temp_user.first; in AddVirtualAssignAdd() local
215 if (IsPrimitiveCNode(temp_node, prim::kPrimCast)) { in AddVirtualAssignAdd()
216 temp_node = node_users_map[temp_node].begin()->first; in AddVirtualAssignAdd()
218 if (IsPrimitiveCNode(temp_node, prim::kPrimMirrorMicroStep) || in AddVirtualAssignAdd()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 291 auto temp_node = cnode; in CreateOpInfo() local
295 temp_node = GraphOutNode(output, tuple_index); in CreateOpInfo()
297 if (!IsPipelineCareNode(temp_node)) { in CreateOpInfo()
298 MS_LOG(EXCEPTION) << "Node: " << temp_node->DebugString() << " is not a Pipeline Care Node."; in CreateOpInfo()
300 if (IsPrimitiveCNode(temp_node, prim::kPrimVirtualDataset)) { in CreateOpInfo()
301 SetVirtualDatasetStrategy(temp_node); in CreateOpInfo()
303 auto shape_list = ExtractShape(temp_node); in CreateOpInfo()
305 MS_LOG(EXCEPTION) << "Node: " << temp_node->DebugString() << " failed to extract shape."; in CreateOpInfo()
307 auto prim = GetValueNode<PrimitivePtr>(temp_node->input(0)); in CreateOpInfo()
310 MS_LOG(EXCEPTION) << "Reshape op can't be a border. node:" << temp_node->DebugString(); in CreateOpInfo()
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fisson/ |
| D | eliminate_concat_split.cc | 44 auto temp_node = cnode->input(1); in GetRealPrevCNode() local
45 if (temp_node == nullptr) { in GetRealPrevCNode()
49 return GetRealPrevCNode(temp_node); in GetRealPrevCNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | eliminate_redundant_op.cc | 47 auto temp_node = cnode->input(index + IntToSize(1)); in GetRealPrevCNode() local
48 MS_EXCEPTION_IF_NULL(temp_node); in GetRealPrevCNode()
50 return GetRealPrevCNode(temp_node, 0, pass_vector); in GetRealPrevCNode()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/ |
| D | function_block.cc | 301 AnfNodePtr temp_node = prev->ReadVariable(var); in SearchReplaceNode() local
302 MS_EXCEPTION_IF_NULL(temp_node); in SearchReplaceNode()
303 if (temp_node != phi) { in SearchReplaceNode()
305 arg_node = temp_node; in SearchReplaceNode()
308 } else if (temp_node == arg_node) { in SearchReplaceNode()
314 << ", node2: " << temp_node->DebugString(); in SearchReplaceNode()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/ |
| D | transpose_transdata_fusion_test.cc | 129 auto temp_node = temp->input(1)->cast<CNodePtr>(); in TEST_F() local
130 MS_LOG(INFO) << "input(11) name:" << temp_node->fullname_with_scope(); in TEST_F()
|