GetActualKernelType => GetExpectedKernelType

add_depthwiseConv_op_gpu
xuwei06 7 years ago
parent ce233796ea
commit 7306aab61d

@ -33,7 +33,7 @@ class AssignValueOp : public framework::OperatorWithKernel {
}
protected:
framework::OpKernelType GetActualKernelType(
framework::OpKernelType GetExpectedKernelType(
const framework::ExecutionContext &ctx) const override {
return framework::OpKernelType(
framework::proto::DataType(ctx.Attr<int>("dtype")), ctx.GetPlace());

Loading…
Cancel
Save