parent
1ed237c11e
commit
5847b96a61
@ -1,5 +1,5 @@
|
||||
#include <paddle/framework/op_registry.h>
|
||||
#include <paddle/operators/sigmoid_op.h>
|
||||
#include <paddle/operators/softmax_op.h>
|
||||
|
||||
REGISTER_OP_GPU_KERNEL(
|
||||
softmax, paddle::operators::SoftmaxKernel<paddle::platform::GPUPlace>);
|
||||
|
Loading…
Reference in new issue