register fp16 bf16 for shape64 cpu kernel (#71764)

Author: ooooo-create

paddle/phi/kernels/shape_kernel.cc

@@ -137,7 +137,9 @@ PD_REGISTER_KERNEL(shape64,
                    float,
                    double,
                    phi::dtype::complex<float>,
-                   phi::dtype::complex<double>) {
+                   phi::dtype::complex<double>,
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16) {
   kernel->InputAt(0).SetBackend(phi::Backend::ALL_BACKEND);
   kernel->OutputAt(0).SetBackend(phi::Backend::CPU);
   kernel->OutputAt(0).SetDataType(phi::DataType::INT64);