cross_entropy_op.cu 193 字节
Newer Older
Q
qijun 已提交
1
#define EIGEN_USE_GPU
Q
Qiao Longfei 已提交
2 3 4
#include "paddle/operators/cross_entropy_op.h"

REGISTER_OP_GPU_KERNEL(onehot_cross_entropy,
5
                       ops::OnehotCrossEntropyOpKernel<ops::GPUPlace, float>);