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