File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -21,7 +21,7 @@ namespace operators {
2121
2222// Out = max(X, 0) - X * Labels + log(1 + exp(-abs(X)))
2323template <typename Place, typename T>
24- class SigmoidCrossEntropyWithLogitsKernel : public framework::OpKernel {
24+ class SigmoidCrossEntropyWithLogitsKernel : public framework::OpKernel<T> {
2525 public:
2626 void Compute(const framework::ExecutionContext &context) const override {
2727 const framework::Tensor *X = context.Input<framework::Tensor>("X");
@@ -48,7 +48,7 @@ class SigmoidCrossEntropyWithLogitsKernel : public framework::OpKernel {
4848
4949// dX = sigmoid(X) - labels
5050template <typename Place, typename T>
51- class SigmoidCrossEntropyWithLogitsGradKernel : public framework::OpKernel {
51+ class SigmoidCrossEntropyWithLogitsGradKernel : public framework::OpKernel<T> {
5252 public:
5353 void Compute(const framework::ExecutionContext &context) const override {
5454 const framework::Tensor *X = context.Input<framework::Tensor>("X");
You can’t perform that action at this time.
0 commit comments