Searched defs:reduce_node (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | bce_with_logits_loss_fusion.cc | 56 auto reduce_node = func_graph->NewCNode(reduce_inputs); in AddReduceNode() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | bce_with_logits_loss_fission.cc | 58 auto reduce_node = func_graph->NewCNode(reduce_inputs); in AddReduceNode() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | sparse_softmax_cross_entropy_with_logits_unify_mindir.cc | 185 auto reduce_node = graph->NewCNode(inputs); in CreateReduceMean() local
415 auto reduce_node = CreateReduceMean(graph, sparse_softmax_node, softmax_node_outputs[0]); in Process() local
452 auto reduce_node = CreateReduceMean(graph, sparse_softmax_node_grad, softmax_node_outputs[0]); in Process() local
504 auto reduce_node = CreateReduceMean(graph, sparse_softmax_node_grad, softmax_node_outputs[0]); in Process() local
535 auto reduce_node = CreateReduceMean(graph, sparse_softmax_node, softmax_node_outputs[0], true); in Process() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | raise_reduction_precision.cc | 61 void RaiseReductionPrecision::ReplaceNode(const AnfNodePtr &reduce_node, const AnfNodePtr &cast_nod… in ReplaceNode() argument
|
| /third_party/mindspore/tests/ut/cpp/optimizer/ |
| D | lib_test.cc | 284 auto reduce_node = inputs[0]; in TEST_F() local
306 auto reduce_node = inputs[0]; in TEST_F() local
327 auto reduce_node = inputs[0]; in TEST_F() local
|