cross_entropy_op.cu 271 字节
Newer Older
Q
Qiao Longfei 已提交
1 2 3 4 5 6
#include "paddle/operators/cross_entropy_op.h"
#include "paddle/framework/op_registry.h"

REGISTER_OP_GPU_KERNEL(onehot_cross_entropy,
                       paddle::operators::OnehotCrossEntropyOpKernel<
                            ::paddle::platform::GPUPlace, float>);