|
|
|
@ -70,8 +70,9 @@ OnehotCrossEntropy Operator.
|
|
|
|
|
namespace ops = paddle::operators;
|
|
|
|
|
REGISTER_OP(onehot_cross_entropy, ops::OnehotCrossEntropyOp,
|
|
|
|
|
ops::OnehotCrossEntropyOpMaker);
|
|
|
|
|
REGISTER_OP_CPU_KERNEL(onehot_cross_entropy,
|
|
|
|
|
ops::OnehotCrossEntropyOpKernel<paddle::platform::CPUPlace, float>);
|
|
|
|
|
REGISTER_OP_CPU_KERNEL(
|
|
|
|
|
onehot_cross_entropy,
|
|
|
|
|
ops::OnehotCrossEntropyOpKernel<paddle::platform::CPUPlace, float>);
|
|
|
|
|
REGISTER_GRADIENT_OP(onehot_cross_entropy, onehot_cross_entropy_grad,
|
|
|
|
|
ops::OnehotCrossEntropyGradientOp);
|
|
|
|
|
REGISTER_OP_CPU_KERNEL(
|
|
|
|
|