Searched defs:reduce_node (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/ |
| D | bce_with_logits_loss_fission.cc | 66 auto reduce_node = NewCNode(reduce_inputs, func_graph); in AddReduceNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/ |
| D | raise_reduction_precision.cc | 65 void RaiseReductionPrecision::ReplaceNode(const AnfNodePtr &reduce_node, const AnfNodePtr &cast_nod… in ReplaceNode() argument
|
| D | tensor_inplace.cc | 90 auto reduce_node = std::find_if(nodes.begin(), nodes.end(), [](const AnfNodePtr &node) { in ReduceChecker() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | sparse_softmax_cross_entropy_with_logits_unify_mindir.cc | 254 auto reduce_node = pass.NewCNode(inputs, graph); in CreateReduceMean() local
572 auto reduce_node = CreateReduceMean(graph, sparse_softmax_node, softmax_node_outputs[0], *this); in Process() local
617 …auto reduce_node = CreateReduceMean(graph, sparse_softmax_node_grad, softmax_node_outputs[0], *thi… in Process() local
685 …auto reduce_node = CreateReduceMean(graph, sparse_softmax_node_grad, softmax_node_outputs[0], *thi… in Process() local
728 auto reduce_node = CreateReduceMean(graph, sparse_softmax_node, softmax_node_outputs[0], *this); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/pass/online_fusion/ |
| D | split_reduce_concat_fusion_pass.cc | 53 auto &reduce_node = node_list_.at(reduce_node1_index); in SatifyReduceConcatParse() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/optimizer/ |
| D | lib_test.cc | 276 auto reduce_node = inputs[0]; in TEST_F() local
298 auto reduce_node = inputs[0]; in TEST_F() local
319 auto reduce_node = inputs[0]; in TEST_F() local
|