IndicateDataType --> GetKernelType

mobile_baidu
yangyaming 8 years ago
parent 40367d18d4
commit 3c84ebec62

@ -45,9 +45,10 @@ class ChunkEvalOp : public framework::OperatorWithKernel {
}
protected:
framework::DataType IndicateDataType(
framework::OpKernelType GetKernelType(
const framework::ExecutionContext &ctx) const override {
return framework::DataType::FP32;
return framework::OpKernelType(framework::DataType::FP32,
ctx.device_context());
}
};

Loading…
Cancel
Save