PaddlePaddle / Paddle
10 个月前同步成功

代码
- 文件
- 提交
- 分支
- Tags
- 贡献者
- 分支图
- Diff
Issue 1423
- 列表
- 看板
- 标记
- 里程碑
合并请求 543
Wiki 0
- Wiki
分析
- 仓库
- DevOps
项目成员
Pages

前往新版Gitcode，体验更适合开发者的 AI 搜索 >>

查找文件 Blame 历史永久链接 Permalink

L

[NPU] use SparseSoftmaxCrossEntropyWithLogits in npu kernel of softmax_with_cross_entropy (#32858) · ff825238

由 Leo Chen 提交于 6月 15, 2021

* use SparseSoftmaxCrossEntropyWithLogits

* fix

* test_slice

* revert test_slice

* add backprob for npu kernel

* fix typo

* fix ut

* fix ut

* refine comments

* return softmax

ff825238

softmax_with_cross_entropy_op_npu.cc 5.2 KB