fix

PaddlePaddle · Apr 6, 2023 · 3cdf9e8 · 3cdf9e8
1 parent e49e1c2
commit 3cdf9e8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/paddle/phi/ops/compat/conv_transpose_sig.cc b/paddle/phi/ops/compat/conv_transpose_sig.cc
@@ -123,6 +123,9 @@ KernelSignature DepthwiseConv2dTransposeGradOpArgumentMapping(
 
 }  // namespace phi
 
+PD_REGISTER_BASE_KERNEL_NAME(conv2d_transpose_grad_grad,
+                             conv2d_transpose_double_grad);
+
 PD_REGISTER_ARG_MAPPING_FN(conv2d_transpose,
                            phi::Conv2dTransposeOpArgumentMapping);
 PD_REGISTER_ARG_MAPPING_FN(conv2d_transpose_grad,