refine fill with tensor (#56568)

paddle/phi/kernels/cpu/full_kernel.cc

@@ -147,4 +147,7 @@ PD_REGISTER_KERNEL(full_with_tensor,
                    phi::dtype::float16,
                    phi::dtype::bfloat16,
                    phi::dtype::complex<float>,
-                   phi::dtype::complex<double>) {}
+                   phi::dtype::complex<double>) {
+  kernel->InputAt(0).SetBackend(phi::Backend::CPU);
+  kernel->InputAt(1).SetBackend(phi::Backend::CPU);
+}

paddle/phi/kernels/gpu/full_kernel.cu

@@ -162,4 +162,7 @@ PD_REGISTER_KERNEL(full_with_tensor,
                    phi::dtype::float16,
                    phi::dtype::bfloat16,
                    phi::dtype::complex<float>,
-                   phi::dtype::complex<double>) {}
+                   phi::dtype::complex<double>) {
+  kernel->InputAt(0).SetBackend(phi::Backend::CPU);
+  kernel->InputAt(1).SetBackend(phi::Backend::CPU);
+}

paddle/phi/kernels/selected_rows/full_kernel.cc

@@ -109,7 +109,10 @@ PD_REGISTER_KERNEL(full_with_tensor_sr,
                    phi::dtype::float16,
                    phi::dtype::bfloat16,
                    phi::dtype::complex<float>,
-                   phi::dtype::complex<double>) {}
+                   phi::dtype::complex<double>) {
+  kernel->InputAt(0).SetBackend(phi::Backend::CPU);
+  kernel->InputAt(1).SetBackend(phi::Backend::CPU);
+}
 
 #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
 PD_REGISTER_KERNEL(full_with_tensor_sr,
@@ -125,7 +128,10 @@ PD_REGISTER_KERNEL(full_with_tensor_sr,
                    bool,
                    phi::dtype::float16,
                    phi::dtype::complex<float>,
-                   phi::dtype::complex<double>) {}
+                   phi::dtype::complex<double>) {
+  kernel->InputAt(0).SetBackend(phi::Backend::CPU);
+  kernel->InputAt(1).SetBackend(phi::Backend::CPU);
+}
 #endif
 
 #if defined(PADDLE_WITH_XPU)
@@ -139,5 +145,8 @@ PD_REGISTER_KERNEL(full_with_tensor_sr,
                    int,
                    int64_t,
                    bool,
-                   phi::dtype::float16) {}
+                   phi::dtype::float16) {
+  kernel->InputAt(0).SetBackend(phi::Backend::CPU);
+  kernel->InputAt(1).SetBackend(phi::Backend::CPU);
+}
 #endif

paddle/phi/kernels/xpu/full_kernel.cc

@@ -165,4 +165,7 @@ PD_REGISTER_KERNEL(full_with_tensor,
                    int,
                    int64_t,
                    bool,
-                   phi::dtype::float16) {}
+                   phi::dtype::float16) {
+  kernel->InputAt(0).SetBackend(phi::Backend::CPU);
+  kernel->InputAt(1).SetBackend(phi::Backend::CPU);
+}