Home
last modified time | relevance | path

Searched defs:softmax_cnode (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_megatron_op_adjust.cc150 auto softmax_cnode = func_graph->NewCNode(softmax_prim_c, {where_cnode}); in AdjustMegatronScaledMaskedSoftmax() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dflash_attention_fusion_for_custom.cc405 auto softmax_cnode = cast1_cnode->input(1)->cast<CNodePtr>(); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dslice_prepose_pass.cc847 const CNodePtr &softmax_cnode) { in PreposeWithSoftmax()