IndicateDataType --> GetKernelType

mobile_baidu
yangyaming 8 years ago
parent 40367d18d4
commit 3c84ebec62

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

Loading…
Cancel
Save