Searched defs:softmax_sum (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | flash_attention_score_grad.cc | 34 const std::optional<BaseTensorPtr> &softmax_max, const std::optional<BaseTensorPtr> &softmax_sum, in FlashAttentionScoreGradAscendCall()
88 const std::optional<BaseTensorPtr> &softmax_max, const std::optional<BaseTensorPtr> &softmax_sum, in FlashAttentionScoreGradAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ |
| D | flash_sp.cc | 387 … const CNodePtr &softmax_max, const CNodePtr &softmax_sum, CNodePtr attention_output, in UpdateAttentionOutput()
460 …CNodePtr local_fa_node, kv_received_tuple, softmax_max, softmax_sum, softmax_out, attention_output; in CreateReplaceFSPGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | flash_attention_score_info.cc | 1296 auto softmax_sum = in ComputeReplaceGraphForLoadBalance() local
|