| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | reorder_ops.cc | 89 void SetNodeInfo(const CNodePtr &orig_node, const CNodePtr &new_node, const NodeIOInfo &node_io_inf… in SetNodeInfo() argument
90 MS_EXCEPTION_IF_NULL(orig_node); in SetNodeInfo()
94 auto orig_node_name = AnfAlgo::GetCNodeName(orig_node); in SetNodeInfo()
101 …MS_LOG(EXCEPTION) << "Can not set empty output type of new node from " << orig_node->fullname_with… in SetNodeInfo()
110 MS_EXCEPTION_IF_NULL(orig_node->abstract()); in SetNodeInfo()
112 orig_node->abstract()->BuildShape()); in SetNodeInfo()
118 AnfAlgo::CopyNodeAttrs(orig_node, new_node); in SetNodeInfo()
126 info_builder.SetKernelType(AnfAlgo::GetKernelType(orig_node)); in SetNodeInfo()
127 info_builder.SetOpPattern(AnfAlgo::GetOpPattern(orig_node)); in SetNodeInfo()
128 info_builder.SetFusionType(AnfAlgo::GetFusionType(orig_node)); in SetNodeInfo()
[all …]
|
| D | graph_kernel_splitter.cc | 72 CNodePtr NewRecomputeNode(const AnfNodePtr &orig_node, std::map<AnfNodePtr, AnfNodePtr> *node_map) { in NewRecomputeNode() argument
73 auto func_graph = orig_node->func_graph(); in NewRecomputeNode()
75 auto cnode = orig_node->cast<CNodePtr>(); in NewRecomputeNode()
92 ScopePtr scope = (orig_node->scope() != kDefaultScope) ? orig_node->scope() : kDefaultScope; in NewRecomputeNode()
97 (*node_map)[orig_node] = cp_node; in NewRecomputeNode()
172 for (auto orig_node : orig_region_nodes) { in DecodeSplitNodes() local
173 ConnectRecomputeOps(&no_cp_nodes, orig_node, node_map[orig_node]); in DecodeSplitNodes()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | operator_info.cc | 137 AnfNodePtr OperatorInfo::CreateConcateNode(const CNodePtr &orig_node, const std::vector<AnfNodePtr>… in CreateConcateNode() argument
139 MS_EXCEPTION_IF_NULL(orig_node); in CreateConcateNode()
158 concat_cnode->set_scope(orig_node->scope()); in CreateConcateNode()
164 AnfNodePtr OperatorInfo::CreateReduceNode(const CNodePtr &orig_node, const std::vector<AnfNodePtr> … in CreateReduceNode() argument
166 MS_EXCEPTION_IF_NULL(orig_node); in CreateReduceNode()
185 addn_cnode->set_scope(orig_node->scope()); in CreateReduceNode()
|
| D | operator_info.h | 67 …AnfNodePtr CreateConcateNode(const CNodePtr &orig_node, const std::vector<AnfNodePtr> &input_nodes,
69 AnfNodePtr CreateReduceNode(const CNodePtr &orig_node, const std::vector<AnfNodePtr> &input_nodes,
|
| D | conv2d_info.cc | 144 AnfNodePtr Conv2DInfo::CreateOutputsOfSplit(const CNodePtr &orig_node, size_t input_index, in CreateOutputsOfSplit() argument
147 MS_ASSERT(orig_node != nullptr && split_outputs != nullptr); in CreateOutputsOfSplit()
149 auto ori_node_name = orig_node->fullname_with_scope(); in CreateOutputsOfSplit()
191 split_inputs.push_back(orig_node->input(input_index + 1)); in CreateOutputsOfSplit()
|
| D | conv2d_info.h | 44 …AnfNodePtr CreateOutputsOfSplit(const CNodePtr &orig_node, size_t input_index, std::vector<AnfNode…
|
| D | depthwise_conv2d_info.h | 41 …AnfNodePtr CreateOutputsOfSplit(const CNodePtr &orig_node, size_t input_index, std::vector<AnfNode…
|
| D | depthwise_conv2d_info.cc | 334 MS_ASSERT(orig_node != nullptr && split_outputs != nullptr); in CreateOutputsOfSplit()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 156 …NodePtr InsertTransOpForMultipleOutput(const FuncGraphPtr &func_graph, const AnfNodePtr &orig_node, in InsertTransOpForMultipleOutput() argument
162 auto update_states = AnfAlgo::GetUpdateStateUsers(manager, orig_node); in InsertTransOpForMultipleOutput()
368 AnfNodePtr InsertTransOpForOutput(const FuncGraphPtr &func_graph, const AnfNodePtr &orig_node, cons… in InsertTransOpForOutput() argument
384 return InsertTransOpForMultipleOutput(func_graph, orig_node, node, kernel_select); in InsertTransOpForOutput()
|
| D | ascend_helper.h | 118 AnfNodePtr InsertTransOpForOutput(const FuncGraphPtr &func_graph, const AnfNodePtr &orig_node, cons…
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/ |
| D | function_block.h | 77 const AnfNodePtr &local_dict_node, const AnfNodePtr &orig_node);
|
| D | function_block.cc | 261 … const AnfNodePtr &local_dict_node, const AnfNodePtr &orig_node) { in MakeInterpret() argument
267 node->set_interpreted_node(orig_node); in MakeInterpret()
|
| /third_party/mesa3d/src/gallium/drivers/lima/ir/gp/ |
| D | scheduler.c | 1074 static bool try_spill_nodes(sched_ctx *ctx, gpir_node *orig_node) in try_spill_nodes() argument
1085 if (node == orig_node) in try_spill_nodes()
1109 if (node == orig_node) in try_spill_nodes()
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 1884 std::string OnnxExporter::GetNodeInputName(const AnfNodePtr &orig_node, std::map<AnfNodePtr, size_t… in GetNodeInputName() argument
1886 auto node = GetRealInput(orig_node); in GetNodeInputName()
|