Home
last modified time | relevance | path

Searched defs:real_div_node (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dsparse_softmax_cross_entropy_with_logits_unify_mindir.cc265 void UpdateAbstract(const CNodePtr &real_div_node, const CNodePtr &expand_dims_node) { in UpdateAbstract()
274 CNodePtr CreateExpandDims(const FuncGraphPtr &graph, const CNodePtr &real_div_node, const PatternPr… in CreateExpandDims()
302 CNodePtr CreateExpandDimsPynative(const FuncGraphPtr &graph, const CNodePtr &real_div_node, in CreateExpandDimsPynative()
408 auto real_div_node = pass.NewCNode(real_div_inputs, graph); in CreateRealDiv() local
526 CNodePtr real_div_node; in CreateMulInput() local
619 CNodePtr real_div_node; in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/function/
Dfunc_pass.cc197 NodePtr CreateExpandDims(const NodePtr &real_div_node) { in CreateExpandDims()
217 NodePtr real_div_node; in CreateMulInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/
Dir_pass.cc359 auto real_div_node = ir_bprop_->ad_param()->tape_->FuncGraph::NewCNode(real_div_inputs); in CreateRealDiv() local
365 CNodePtr CreateExpandDims(const CNodePtr &real_div_node) { in CreateExpandDims()
402 CNodePtr real_div_node; in CreateMulInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dcame_parallel_handler.cc455 auto real_div_node = func_graph->NewCNode(real_div_input); in InsertAllReduceAndRealDivToReduceMeanInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgraph_splitter.cc787 const auto &real_div_node = grad_accum_nodes.second; in ProcessForSplitOptimizer() local