File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -53,7 +53,7 @@ class AucOp : public framework::OperatorWithKernel {
53
53
const framework::ExecutionContext &ctx) const override {
54
54
return framework::OpKernelType (
55
55
framework::ToDataType (ctx.Input <Tensor>(" Predict" )->type ()),
56
- ctx. device_context ());
56
+ platform::CPUPlace ());
57
57
}
58
58
};
59
59
Original file line number Diff line number Diff line change @@ -69,7 +69,7 @@ class NCEOp : public framework::OperatorWithKernel {
69
69
const framework::ExecutionContext& ctx) const override {
70
70
return framework::OpKernelType (
71
71
framework::ToDataType (ctx.Input <Tensor>(" Input" )->type ()),
72
- ctx. GetPlace ());
72
+ platform::CPUPlace ());
73
73
}
74
74
};
75
75
@@ -174,7 +174,7 @@ class NCEOpGrad : public framework::OperatorWithKernel {
174
174
const framework::ExecutionContext& ctx) const override {
175
175
return framework::OpKernelType (
176
176
framework::ToDataType (ctx.Input <Tensor>(" Input" )->type ()),
177
- ctx. GetPlace ());
177
+ platform::CPUPlace ());
178
178
}
179
179
};
180
180
You can’t perform that action at this time.
0 commit comments