Searched defs:expand_dims_node (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | sparse_softmax_cross_entropy_with_logits_unify_mindir.cc | 265 void UpdateAbstract(const CNodePtr &real_div_node, const CNodePtr &expand_dims_node) { in UpdateAbstract()
294 auto expand_dims_node = pass.NewCNode(expand_dims_inputs, graph); in CreateExpandDims() local
315 auto expand_dims_node = pass.NewCNode(expand_dims_inputs, graph); in CreateExpandDimsPynative() local
532 auto expand_dims_node = CreateExpandDimsPynative(graph, real_div_node, pass); in CreateMulInput() local
625 auto expand_dims_node = CreateExpandDims(graph, real_div_node, *this); in Process() local
763 auto expand_dims_node = in Process() local
825 auto expand_dims_node = in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/function/ |
| D | func_pass.cc | 70 auto expand_dims_node = CreateMulInput(inputs, dout, &softmax_node_outputs); in Run() local
223 auto expand_dims_node = CreateExpandDims(real_div_node); in CreateMulInput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/ |
| D | ir_pass.cc | 172 auto expand_dims_node = CreateMulInput(mul_node, sparse_softmax_node, &softmax_node_outputs); in Run() local
382 auto expand_dims_node = ir_bprop_->ad_param()->tape_->FuncGraph::NewCNode(expand_dims_inputs); in CreateExpandDims() local
409 auto expand_dims_node = CreateExpandDims(real_div_node); in CreateMulInput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.cc | 3335 auto expand_dims_node = FindExpandDimsWIthGradScale(cnode, node_user_map, bfs_depth); in InsertDivAndAllReduceForNorm() local
|