parent
8ad9006dd5
commit
736d078cbf
@ -1,5 +1,5 @@
|
||||
#include <paddle/operators/sigmoid_op.h>
|
||||
#include <paddle/framework/op_registry.h>
|
||||
#include "paddle/operators/sigmoid_op.h"
|
||||
#include "paddle/framework/op_registry.h"
|
||||
|
||||
REGISTER_OP_GPU_KERNEL(
|
||||
sigmoid, paddle::operators::SigmoidKernel<paddle::platform::GPUPlace, float>);
|
||||
|
@ -1,5 +1,5 @@
|
||||
#include <paddle/framework/op_registry.h>
|
||||
#include <paddle/operators/softmax_op.h>
|
||||
#include "paddle/framework/op_registry.h"
|
||||
#include "paddle/operators/softmax_op.h"
|
||||
|
||||
REGISTER_OP_GPU_KERNEL(
|
||||
softmax, paddle::operators::SoftmaxKernel<paddle::platform::GPUPlace, float>);
|
||||
|
Loading…
Reference in new issue