Searched refs:real_node (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | insert_tensor_move_for_hccl_op.cc | 35 auto real_node = kernel_with_index.first; in IsParameterOrValueNode() local
36 MS_EXCEPTION_IF_NULL(real_node); in IsParameterOrValueNode()
37 if (real_node->isa<Parameter>()) { in IsParameterOrValueNode()
40 return real_node->isa<ValueNode>(); in IsParameterOrValueNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | pattern_engine.cc | 187 auto real_node = utils::cast<AnfNodePtr>(expr_ref); in UpdateEquivMap() local
188 MS_EXCEPTION_IF_NULL(real_node); in UpdateEquivMap()
189 if (!real_node->isa<CNode>()) { in UpdateEquivMap()
205 (*equiv)[iter->second] = real_node; in UpdateEquivMap()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | control_node_parser.cc | 486 …const auto &real_node = AnfAlgo::VisitKernelWithReturnType(node, 0, false, {prim::kPrimTupleGetIte… in FetchRealOutputByCallNode() local
487 if (!IsCallNode(real_node)) { in FetchRealOutputByCallNode()
488 return real_node; in FetchRealOutputByCallNode()
490 if ((*call_nodes).find(real_node) != (*call_nodes).end()) { in FetchRealOutputByCallNode()
493 (void)((*call_nodes).insert(real_node)); in FetchRealOutputByCallNode()
495 const auto &func_graphs = FetchFuncGraphbyCallNode(real_node); in FetchRealOutputByCallNode()
557 const auto &real_node = real_node_with_index.first; in FetchAllRealInputNodeByParameter() local
558 if (real_node->isa<Parameter>()) { in FetchAllRealInputNodeByParameter()
559 if (!HasAbstractRef(real_node) && !HasAbstractMonad(real_node)) { in FetchAllRealInputNodeByParameter()
562 } else if (HasAbstractMonad(real_node)) { in FetchAllRealInputNodeByParameter()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_build.cc | 1281 auto real_node = kernel_idx.first; in GenFusionDataInputJson() local
1285 GenDescJson(real_node, real_idx, real_idx, &output_desc, fusion_data_type); in GenFusionDataInputJson()
1287 auto full_name = real_node->fullname_with_scope(); in GenFusionDataInputJson()
1399 auto real_node = kernel_idx.first; in GenFusionComputeInputJson() local
1402 GenDescJson(real_node, real_idx, real_idx, &input_desc); in GenFusionComputeInputJson()
1537 auto real_node = kernel_idx.first; in CalOutputSize() local
1539 auto full_name = real_node->fullname_with_scope(); in CalOutputSize()
1545 if (output_node != real_node) { in CalOutputSize()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | kernel_graph.cc | 1372 auto real_node = AnfAlgo::VisitKernel(input, 0).first; in SetOptimizerFlag() local
1373 MS_EXCEPTION_IF_NULL(real_node); in SetOptimizerFlag()
1374 if (!real_node->isa<Parameter>()) { in SetOptimizerFlag()
1377 auto param = real_node->cast<ParameterPtr>(); in SetOptimizerFlag()
|