Home
last modified time | relevance | path

Searched refs:CNodePtr (Results 1 – 25 of 1420) sorted by relevance

12345678910>>...57

/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dslice_prepose_pass.h43 static void ClearCNodeAbstractValue(const CNodePtr &cnode);
44 static STATUS SwapSliceWithPreceed(const FuncGraphPtr &graph, const CNodePtr &slice_cnode,
45 … const CNodePtr &preceed_cnode, int index, const TransactionPtr &tr = nullptr);
47 static ValueNodePtr CopySliceValueNode(const CNodePtr &slice_cnode);
48 static CNodePtr InsertSlice(const FuncGraphPtr &graph, const std::vector<AnfNodePtr> &inputs,
49 const CNodePtr &preceed_cnode, int index, const TransactionPtr &tr);
50 static STATUS VerifySliceAttrs(const CNodePtr &slice_cnode, int dim = -1);
51 …static STATUS SliceParamDeBroadcast(const CNodePtr &slice_cnode, const std::vector<int64_t> &ref_s…
53 static CNodePtr CreateReshapeCNode(const FuncGraphPtr &graph, const std::vector<int64_t> &shape,
54 … const AbstractBasePtr &abstract, const CNodePtr &preceed_cnode);
[all …]
Ddecrease_transpose_algo.h43 …STATUS InsertPostTransNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode, const std::vecto…
44 …STATUS InsertPreTransNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode, const std::vector…
45 …STATUS GenNewInput(const FuncGraphPtr &func_graph, const CNodePtr &cnode, const std::vector<int> p…
47 bool RunDoFixFormat(const FuncGraphPtr &func_graph, const CNodePtr &cnode);
51 STATUS PostTransposeFusion(const FuncGraphPtr &func_graph, const CNodePtr &cnode);
53 STATUS HandleGraphMultiNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode,
54 std::set<CNodePtr> *visit_transposes);
55 …STATUS InsertPreTransNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode, TransTypePair *tr…
56 int SetSubGraphInput(const CNodePtr &cnode, const FuncGraphPtr &sub_graph);
59 int SetSubGraphAbstract(const CNodePtr &cnode, const FuncGraphPtr &sub_graph);
[all …]
Dcontrol_flow_pass.h47 const CNodePtr &aim_cnode, FuncGraphPtr *after_fg);
51 …const FuncGraphPtr &fg, const CNodePtr &while_cnode, const std::vector<AnfNodePtr> &visited_nodes_…
52 CNodePtr *cond_partial_cnode, std::vector<AnfNodePtr> *cond_nodes_used_by_after_partial,
54 …int CreateWhileBodyPartialNode(const FuncGraphPtr &cond_fg, const CNodePtr &while_cnode, CNodePtr …
59 CNodePtr *while_cnode, CNodePtr *after_partial_cnode);
64 int CreateIfPartialNodeExternalInputs(const CNodePtr &if_cnode, const FuncGraphPtr &partial_fg,
67 … std::vector<AnfNodePtr> *fg_inputs_only_used_by_after_partial, CNodePtr *if_cnode,
68 FuncGraphPtr *after_fg, CNodePtr *then_partial_cnode);
70 … std::vector<AnfNodePtr> *fg_inputs_only_used_by_after_partial, CNodePtr *if_cnode,
71 FuncGraphPtr *after_fg, CNodePtr *then_partial_cnode);
[all …]
Ddecrease_transpose_algo.cc31 STATUS FindAreaSurroundedByTranspose(const FuncGraphPtr &func_graph, const CNodePtr &root_node, in FindAreaSurroundedByTranspose()
32 std::set<CNodePtr> *in_nodes, std::set<CNodePtr> *out_nodes, in FindAreaSurroundedByTranspose()
33 std::set<CNodePtr> *middle_nodes) { in FindAreaSurroundedByTranspose()
36 std::queue<CNodePtr> queue_nodes{}; in FindAreaSurroundedByTranspose()
62 if (!utils::isa<CNodePtr>(cur_node->input(i))) { in FindAreaSurroundedByTranspose()
65 auto cur_node_input = cur_node->input(i)->cast<CNodePtr>(); in FindAreaSurroundedByTranspose()
82 if (!utils::isa<CNodePtr>(cur_node_user.first)) { in FindAreaSurroundedByTranspose()
86 auto cur_node_post = cur_node_user.first->cast<CNodePtr>(); in FindAreaSurroundedByTranspose()
102 void SetTransType(const std::set<CNodePtr> &cnodes, FormatTransNodeType *trans_type) { in SetTransType()
121 bool JudgeCanOptimizerForMultiOp(const std::set<CNodePtr> &in_nodes, const std::set<CNodePtr> &out_… in JudgeCanOptimizerForMultiOp()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_stream_assign.h32 CNodePtr mock_send_node;
33 CNodePtr mock_recv_node;
39 CNodePtr cnode;
56 std::vector<CNodePtr>::iterator FindSendNodePos(std::vector<CNodePtr>::iterator begin,
57 … std::vector<CNodePtr>::iterator end, const CNodePtr mock_recv_node,
61 std::vector<CNodePtr>::iterator FindRecvNodePos(std::vector<CNodePtr>::iterator begin,
62 … std::vector<CNodePtr>::iterator end, const CNodePtr mock_send_node,
67 … const CNodePtr &mock_send_node, const CNodePtr &mock_recv_node, CNodePtr *send_node,
68 CNodePtr *recv_node);
69 CNodePtr CreateStreamSwitchNode(const std::shared_ptr<session::KernelGraph> &kernel_graph, const st…
[all …]
Dgpu_stream_assign.cc32 std::vector<CNodePtr> allreduce_kernels; in AssignGpuStream()
51 …allreduce_kernels.begin(), allreduce_kernels.end(), allreduce_kernels.begin(), [&](CNodePtr allred… in AssignGpuStream()
67 std::vector<CNodePtr>::iterator iter, iter_begin; in FindAllReduceStreamSwitchPos()
69 std::vector<CNodePtr>::iterator iter_end = execution_kernels.end(); in FindAllReduceStreamSwitchPos()
74 std::vector<CNodePtr>::iterator mock_send_node_iter = in FindAllReduceStreamSwitchPos()
86 std::vector<CNodePtr>::iterator mock_recv_node_iter = in FindAllReduceStreamSwitchPos()
109 std::vector<CNodePtr>::iterator FindSendNodePos(std::vector<CNodePtr>::iterator begin, in FindSendNodePos()
110 … std::vector<CNodePtr>::iterator end, const CNodePtr mock_recv_node, in FindSendNodePos()
123 std::vector<CNodePtr>::iterator FindRecvNodePos(std::vector<CNodePtr>::iterator begin, in FindRecvNodePos()
124 … std::vector<CNodePtr>::iterator end, const CNodePtr mock_send_node, in FindRecvNodePos()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_adjust.h68 CNodePtr CreateStreamActiveOp(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr);
74 CNodePtr CreateNPUGetFloatStatus(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
75 const CNodePtr &npu_cnode);
76 CNodePtr CreateNPUClearStatus(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr,
77 const CNodePtr &npu_cnode);
78 CNodePtr CreateNPUAllocStatus(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr);
79 …CNodePtr CreateAssignAdd(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr, const CNod…
81 …CNodePtr CreateAssign(const std::shared_ptr<session::KernelGraph> &kernel_graph_ptr, const AnfNode…
83 …CNodePtr CreateRecvApplyKernel(const std::shared_ptr<session::KernelGraph> &graph_ptr, uint32_t ev…
84 …CNodePtr CreateSendApplyKernel(const std::shared_ptr<session::KernelGraph> &graph_ptr, uint32_t ev…
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Ddeal_ref_and_split_unsupported_transdata.h37 …CNodePtr MakeDependency(const CNodePtr &getitem, const CNodePtr &final_node, const CNodePtr &cnode,
39 …CNodePtr SplitTransdataIfNotSupported(const FuncGraphPtr &func_graph, const CNodePtr &cnode) const;
40 void DealBroadCastAsRef(const FuncGraphPtr &func_graph, const CNodePtr &cnode) const;
41 CNodePtr DealRefSingleOutput(const FuncGraphPtr &func_graph, const CNodePtr &cnode,
43 CNodePtr DealRefForMultipleOutput(const FuncGraphPtr &func_graph, const CNodePtr &cnode,
45 …CNodePtr AddAdditionalToRefOutput(const FuncGraphPtr &func_graph, const CNodePtr &cnode, size_t ou…
46 size_t input_index, const CNodePtr &get_item) const;
47 …void AddRefPairToKernelGraph(const FuncGraphPtr &func_graph, const CNodePtr &cnode, const AnfNodeP…
50 …void AddRefNodePairToKernelGraph(const FuncGraphPtr &func_graph, const CNodePtr &cnode, const size…
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_stream_assign.h116 const std::map<CNodePtr, CNodePtr> &get_event_map() const { return event_map_; } in get_event_map()
122 …CNodePtr CreateSendApplyKernel(const NotNull<KernelGraphPtr> &graph_ptr, uint32_t event_id, uint32…
123 …CNodePtr CreateRecvApplyKernel(const NotNull<KernelGraphPtr> &graph_ptr, uint32_t event_id, uint32…
128 void AssignCommonStreamId(const CNodePtr &cur_cnode_ptr);
130 uint32_t AssignHcomStreamId(const CNodePtr &cur_cnode_ptr, bool new_graph);
132 uint32_t AssignIndependentStreamId(const CNodePtr &cur_cnode_ptr, bool new_graph);
144 bool CheckStreamSwitch(const CNodePtr &switch_ptr);
156 …std::vector<std::pair<uint32_t, vector<size_t>>> GetStreamIDHcomMap(const std::vector<CNodePtr> &c…
160 …vector<CNodePtr> GetLastInputCnode(const NotNull<KernelGraphPtr> &graph_ptr, const CNodePtr &cur_c…
161 …dHcom(const std::vector<std::pair<uint32_t, vector<size_t>>> &hcom_index, const CNodePtr &node_ptr,
[all …]
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_while.h35 …FunctionalizeWhile(std::vector<AnfNodePtr> node_cluster, const CNodePtr &loop_cond_node, FuncGraph… in FunctionalizeWhile()
62 CNodePtr BlongToWhichSwitch(const CNodePtr &node);
63 CNodePtr BlongToWhichMerge(const CNodePtr &node);
64 CNodePtr BlongToWhichEnter(const CNodePtr &node);
65 CNodePtr BlongToWhichExternalEnter(const CNodePtr &node);
66 int PosInInputEnterNodes(const CNodePtr &node);
73 const CNodePtr loop_cond_node_;
78 CNodePtr while_node_ = nullptr;
84 std::vector<CNodePtr> input_enter_nodes_{};
85 std::vector<CNodePtr> external_input_enter_nodes_{};
[all …]
Dfunctionalize_while.cc45 CNodePtr FunctionalizeWhile::BlongToWhichSwitch(const CNodePtr &node) { in BlongToWhichSwitch()
48 CNodePtr FunctionalizeWhile::BlongToWhichMerge(const CNodePtr &node) { in BlongToWhichMerge()
51 CNodePtr FunctionalizeWhile::BlongToWhichEnter(const CNodePtr &node) { in BlongToWhichEnter()
55 CNodePtr FunctionalizeWhile::BlongToWhichExternalEnter(const CNodePtr &node) { in BlongToWhichExternalEnter()
62 CNodePtr aim_node = nullptr; in BlongToWhichExternalEnter()
67 aim_node = utils::cast<CNodePtr>(input_node); in BlongToWhichExternalEnter()
78 aim_node = utils::cast<CNodePtr>(todo_node); in BlongToWhichExternalEnter()
82 if (utils::isa<CNodePtr>(todo_node)) { in BlongToWhichExternalEnter()
83 auto cnode = utils::cast<CNodePtr>(todo_node); in BlongToWhichExternalEnter()
97 int FunctionalizeWhile::PosInInputEnterNodes(const CNodePtr &node) { in PosInInputEnterNodes()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dupdatestate_eliminate.cc54 bool OnlyUsedByOneNode(const AnfNodePtr &be_used_node, const CNodePtr &given_node) { in OnlyUsedByOneNode()
91 bool CheckHasMonadInput(const CNodePtr &cnode) { in CheckHasMonadInput()
102 for (auto &input : first_input_node->cast<CNodePtr>()->inputs()) { in CheckHasMonadInput()
115 AnfNodePtr NewUpdateStateWithAttach(const CNodePtr &update_state, const AnfNodePtr &attach) { in NewUpdateStateWithAttach()
127 AnfNodePtr EliminateUpdateStateWithDepend(const CNodePtr &update_state) { in EliminateUpdateStateWithDepend()
128 auto depend = update_state->input(kAttachIndex)->cast<CNodePtr>(); in EliminateUpdateStateWithDepend()
170 …dePtr EliminateUpdateStateMakeTupleWithUselessNode(const CNodePtr &update_state, const CNodePtr &m… in EliminateUpdateStateMakeTupleWithUselessNode()
197 void GetLoadsFollowLoad(const CNodePtr &update_state, const CNodePtr &load, std::vector<CNodePtr> *…
198 std::vector<CNodePtr> *loads);
199 void GetLoadsFollowTuple(const CNodePtr &update_state, const CNodePtr &make_tuple, std::vector<CNod…
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.h47 CNodePtr loss_node = nullptr;
58 void ForwardCommunication(OperatorVector forward_op, const CNodePtr &node);
60 void InsertRedistribution(const RedistributionOpListPtr &redistribution_oplist_ptr, const CNodePtr …
61 const FuncGraphPtr &func_graph, int64_t pos, const CNodePtr &pre_node);
63 TensorLayout GetTensorInLayout(const CNodePtr &pre_node, const PrimitivePtr &pre_prim,
66 OperatorInfoPtr GetDistributeOperator(const CNodePtr &node);
69 … const CNodePtr &middle_node, int64_t index, TensorRedistribution tensor_redistribution,
70 const CNodePtr &pre_node);
78 void StepRedistribution(const CNodePtr &node, const OperatorInfoPtr &distribute_operator, const CNo…
79 … const TensorRedistribution &tensor_redistribution, const CNodePtr &pre_node);
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Drecompute.cc35 bool CanNotRecomputed(const CNodePtr &node) { in CanNotRecomputed()
62 auto cnode = node->cast<CNodePtr>(); in GetRecomputeCNodeAttr()
79 bool IsCandidateRecomputedNode(const CNodePtr &node) { return !IsBpropNode(node) && IsSetRecomputeC… in IsCandidateRecomputedNode()
81 std::vector<CNodePtr> FindCandidateRecomputedNodes(const FuncGraphManagerPtr &mng, in FindCandidateRecomputedNodes()
82 const std::vector<CNodePtr> &cnodes) { in FindCandidateRecomputedNodes()
84 std::vector<CNodePtr> candidate_recomputed_nodes; in FindCandidateRecomputedNodes()
111 void GetMaxSubGraph(const FuncGraphManagerPtr &mng, std::unordered_set<CNodePtr> *recomputed_nodes,… in GetMaxSubGraph()
115 std::queue<CNodePtr> nodes_to_visit; in GetMaxSubGraph()
132 auto input_cnode = input->cast<CNodePtr>(); in GetMaxSubGraph()
151 auto output_cnode = output_node->cast<CNodePtr>(); in GetMaxSubGraph()
[all …]
/third_party/mindspore/mindspore/lite/tools/common/
Dfunc_graph_subgraph.h40 int Init(const std::set<CNodePtr> &head_nodes = {});
41 int Reset(const std::set<CNodePtr> &nodes, const std::set<CNodePtr> &head_nodes = {});
45 std::set<CNodePtr> GetNodes() const;
46 std::set<CNodePtr> GetInCNodes() const;
47 std::set<CNodePtr> GetOutCNodes() const;
52 std::set<CNodePtr> GetInputCNodes() const;
53 std::set<CNodePtr> GetOutputCNodes() const;
55 int InitSubGraphNode(const std::set<CNodePtr> &head_nodes);
59 std::set<CNodePtr> FindCommonOutputs(const SubGraphPtr &subgraph) const;
69 static int SetFuncGraphOutput(const FuncGraphPtr &graph, const std::set<CNodePtr> &outputs);
[all …]
Dfunc_graph_subgraph.cc31 int SubGraph::Init(const std::set<CNodePtr> &head_nodes) { in Init()
50 int SubGraph::Reset(const std::set<CNodePtr> &nodes, const std::set<CNodePtr> &head_nodes) { in Reset()
70 std::set<CNodePtr> SubGraph::GetNodes() const { return this->nodes_; } in GetNodes()
72 std::set<CNodePtr> SubGraph::GetInCNodes() const { return this->in_nodes_; } in GetInCNodes()
74 std::set<CNodePtr> SubGraph::GetInputCNodes() const { in GetInputCNodes()
75 std::set<CNodePtr> inputs; in GetInputCNodes()
86 std::set<CNodePtr> SubGraph::GetOutCNodes() const { return this->out_nodes_; } in GetOutCNodes()
88 std::set<CNodePtr> SubGraph::FindCommonOutputs(const SubGraphPtr &subgraph) const { in FindCommonOutputs()
92 std::set<CNodePtr> outputs_this = this->GetOutputCNodes(); in FindCommonOutputs()
96 std::set<CNodePtr> outputs_other = subgraph->GetOutputCNodes(); in FindCommonOutputs()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/allreduce_fusion/
Dallreduce_graph.h45 Status AddNode(const CNodePtr &node, const AnfNodePtr &para);
46 Status AddEdge(const CNodePtr &from, const CNodePtr &to, double dist);
47 bool NodeInGraph(const CNodePtr &node) const;
63 const std::unordered_set<CNodePtr> &cnode_set() const { return cnode_set_; } in cnode_set()
64 CNodePtr head_cnode() const { return head_cnode_; } in head_cnode()
65 Status set_head_cnode(const CNodePtr &node);
69 CNodePtr head_cnode_;
72 std::unordered_set<CNodePtr> cnode_set_;
74 std::unordered_map<AnfNodePtr, std::vector<CNodePtr>> para_cnode_map_;
76 std::unordered_map<AnfNodePtr, std::unordered_set<CNodePtr>> para_cnodeset_map_;
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dkernel_graph.h125 CNodePtr NewCNode(const std::vector<AnfNodePtr> &inputs) override;
126 …CNodePtr NewCNodeWithInfos(const std::vector<AnfNodePtr> &inputs, const CNodePtr &ori_cnode = null…
127 void CreateKernelInfoFromNewParameter(const CNodePtr &cnode);
128 CNodePtr NewCNode(const CNodePtr &cnode);
129 void ResetAssignInputFeatureMapFlag(const CNodePtr &cnode) const;
137 void set_execution_order(const std::vector<CNodePtr> &order) { execution_order_ = order; } in set_execution_order()
138 void set_execution_order(std::vector<CNodePtr> &&order) { execution_order_ = std::move(order); } in set_execution_order()
139 const std::vector<CNodePtr> &execution_order() const { return execution_order_; } in execution_order()
141 …void set_mem_reuse_exec_order(const std::vector<CNodePtr> &order) { mem_reuse_exec_order_ = order;… in set_mem_reuse_exec_order()
142 const std::vector<CNodePtr> &mem_reuse_exec_order() const { return mem_reuse_exec_order_; } in mem_reuse_exec_order()
[all …]
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Deliminate_redundant_op_test.cc57 void SelectKernel(const CNodePtr &cnode) override { in SelectKernel()
91 EXPECT_NE(ret->input(1)->cast<CNodePtr>()->input(1), nullptr); in TEST_F()
93 auto tuple = ret->input(1)->cast<CNodePtr>()->input(1); in TEST_F()
95 EXPECT_NE(tuple->cast<CNodePtr>()->input(1), nullptr); in TEST_F()
97 auto sub = tuple->cast<CNodePtr>()->input(1); in TEST_F()
99 EXPECT_NE(sub->cast<CNodePtr>()->input(1), nullptr); in TEST_F()
100 auto add = sub->cast<CNodePtr>()->input(1); in TEST_F()
158 EXPECT_NE(ret->input(1)->cast<CNodePtr>()->input(1), nullptr); in TEST_F()
160 auto tuple = ret->input(1)->cast<CNodePtr>()->input(1); in TEST_F()
162 EXPECT_NE(tuple->cast<CNodePtr>()->input(1), nullptr); in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dpynative_dfunctor.h30 static ValueNodePtr GenNewTensor(const CNodePtr &forward_node);
32 …static void GetForwardOutNodeAndBpropGraph(const CNodePtr &k_app, CNodePtr *forward_node, FuncGrap…
34 …static std::vector<AnfNodePtr> RunOutputReplace(const CNodePtr &forward_node, const FuncGraphPtr &…
35 … const FuncGraphPtr &fprop_graph, const CNodePtr &cnode_morph);
37 const CNodePtr &cnode_morph);
38 static void ReplaceEquivdout(const CNodePtr &k_app, const CNodePtr &cnode_morph);
Ddfunctor.h79 void BackPropagateSwitchLayer(const CNodePtr &cnode_morph, const CNodePtr &env);
80 …void BackPropagate(const CNodePtr &cnode_morph, const CNodePtr &k_app, const AdjointPtr &node_adjo…
84 AnfNodePtr MapPrimitiveToK(const CNodePtr &primitive_user, size_t index);
97 bool AllReferencesStopped(const CNodePtr &node);
104 ValueNodePtr GenNewTensor(const CNodePtr &forward_node);
106 …void GetForwardOutNodeAndBpropGraph(const CNodePtr &k_app, CNodePtr *forward_node, FuncGraphPtr *b…
108 …std::vector<AnfNodePtr> RunOutputReplace(const CNodePtr &forward_node, const FuncGraphPtr &bprop_g…
109 … const FuncGraphPtr &fprop_graph, const CNodePtr &cnode_morph);
111 const CNodePtr &cnode_morph);
112 void ReplaceEquivdout(const CNodePtr &k_app, const CNodePtr &cnode_morph);
[all …]
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtflite_lstm_cell_fusion.h42 static lite::STATUS SetAbstractTuple(const CNodePtr &cnode, int output_num);
44 …static CNodePtr CreateOutputGetItem(const FuncGraphPtr &func_graph, const CNodePtr &node, int item…
55 static CNodePtr CreateSqueezeNode(const FuncGraphPtr &func_graph, const CNodePtr &input_node,
58 …static lite::STATUS AdjustOtherGetItems(const FuncGraphPtr &func_graph, const CNodePtr &while_cnod…
59 … const CNodePtr &lstm_cnode, const CNodePtr &output_get_item);
65 …virtual CNodePtr CreateLSTMNode(const FuncGraphPtr &func_graph, const EquivPtr &equiv, const Equiv…
69 CNodePtr GetWhileCnode(const AnfNodePtr &cnode) const;
72 static bool CheckReferencedOutputs(const FuncGraphPtr &func_graph, const CNodePtr &while_cnode);
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/
Dop_tiling_adapter.h39 ge::NodePtr AnfNodeToGeNodeAdapter(const CNodePtr &node, ge::ComputeGraphPtr *ge_graph,
44 void ConvertInputShapeAndType(const CNodePtr &node, ge::OpDescPtr *op_desc);
45 void ConvertOutputShapeAndType(const CNodePtr &node, ge::OpDescPtr *op_desc);
46 void ConvertCompileInfo(const CNodePtr &node, ge::OpDescPtr *op_desc);
47 void ConvertAttrs(const CNodePtr &node, ge::OpDescPtr *op_desc);
49 …const CNodePtr &node, const std::map<uint32_t, tensor::TensorPtr> &depend_tensor_map, ge::OpDescPt…
51 …ge::NodePtr NewConstantOp(const CNodePtr &node, const std::string &name, const tensor::TensorPtr &…
55 std::string GetInputName(const CNodePtr &node, size_t index);
56 std::string GetOutputName(const CNodePtr &node, size_t index);
57 void InitOpIoName(const CNodePtr &node);
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dbatch_norm_add_relu_grad_fusion.cc52 auto tuple_getiterm_cnode = output->cast<CNodePtr>(); in GetBatchNormOutputs()
68 void SetShapeAndType(const CNodePtr &bn_add_relu_grad, const AnfNodePtr &bn_grad, const AnfNodePtr … in SetShapeAndType()
84 const CNodePtr &bn_add_relu_grad) { in ReplaceOutput()
129 auto relu_grad = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in PatternCheck()
137 auto tuple_getitem = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 5); in PatternCheck()
139 …if (!utils::isa<CNodePtr>(tuple_getitem) || AnfAlgo::GetCNodeName(tuple_getitem) != prim::kPrimTup… in PatternCheck()
142 auto forward_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tuple_getitem), 0); in PatternCheck()
167 auto relu_grad = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process()
169 auto dy = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(relu_grad), 0); in Process()
171 auto y = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(relu_grad), 1); in Process()
[all …]
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dconvert.h117 void SetOpInput(const OpAdapterPtr &adpt, const CNodePtr &node);
146 AnfNodePtr TraceTupleGetItem(const CNodePtr &node, uint64_t *index);
147 AnfNodePtr TraceMakeTuple(const CNodePtr &node, uint64_t index);
148 AnfNodePtr TraceDepend(const CNodePtr &node);
152 OperatorPtr ConvertCNode(CNodePtr node);
158 void SaveParamFormat(CNodePtr node);
159 void GetCaseNodeInput(const CNodePtr node, const CNodePtr input_node);
160 void ConvertTupleGetItem(const CNodePtr node);
161 void ConvertMakeTuple(const CNodePtr node);
162 void ConvertTopK(const CNodePtr node);
[all …]

12345678910>>...57