|
|
@ -37,7 +37,7 @@ class EditDistanceOp : public framework::OperatorWithKernel {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
protected:
|
|
|
|
protected:
|
|
|
|
framework::OpKernelType GetActualKernelType(
|
|
|
|
framework::OpKernelType GetExpectedKernelType(
|
|
|
|
const framework::ExecutionContext &ctx) const override {
|
|
|
|
const framework::ExecutionContext &ctx) const override {
|
|
|
|
return framework::OpKernelType(framework::proto::DataType::FP32,
|
|
|
|
return framework::OpKernelType(framework::proto::DataType::FP32,
|
|
|
|
ctx.device_context());
|
|
|
|
ctx.device_context());
|
|
|
|