|
|
@ -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());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|