cross_entropy_op.cu 171 字节 编辑Web IDE 1 2 3 4 #include "paddle/operators/cross_entropy_op.h" REGISTER_OP_GPU_KERNEL(onehot_cross_entropy, ops::OnehotCrossEntropyOpKernel<ops::GPUPlace, float>);