Skip to content

the gpu kernel of cross entropy with soft label can be optimized by using reduce. #4236

@lcy-seso

Description

@lcy-seso

The current implementation of the GPU kernel of the cross_entropy op with the soft label can be further optimized by using reduce:
https://github.com/PaddlePaddle/Paddle/blob/develop/paddle/operators/cross_entropy_op.cu#L38

This problem is also known by @qingqing01

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions