[Performance]Set ShapeKernel with ALL_BACKEND and ALL_LAYOUT (#42138)

* [Performance]Set ShapeKernel with ALL_BACKEND and ALL_LAYOUT * [Performance]Set ShapeKernel with ALL_BACKEND and ALL_LAYOUT

[Performance]Set ShapeKernel with ALL_BACKEND and ALL_LAYOUT (#42138)
* [Performance]Set ShapeKernel with ALL_BACKEND and ALL_LAYOUT * [Performance]Set ShapeKernel with ALL_BACKEND and ALL_LAYOUT
79ac8870 · Aurelius84 · GitHub · 1587ad07 · 79ac8870
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

paddle/phi/kernels/shape_kernel.cc paddle/phi/kernels/shape_kernel.cc +3 -1

未找到文件。
--- a/paddle/phi/kernels/shape_kernel.cc
+++ b/paddle/phi/kernels/shape_kernel.cc
@@ -63,5 +63,7 @@ PD_REGISTER_KERNEL(shape,
                   double,
                   phi::dtype::complex<float>,
                   phi::dtype::complex<double>,
-                   phi::dtype::float16) {}
+                   phi::dtype::float16) {
+  kernel->InputAt(0).SetBackend(phi::Backend::ALL_BACKEND);
+}
 #endif