first commit (#45253)

b5d8bd2f · limingshu · GitHub · dd2f4a1c · b5d8bd2f
隐藏空白更改
内联并排

Showing with 3 addition and 6 deletion

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

未找到文件。
--- a/paddle/phi/kernels/shape_kernel.cc
+++ b/paddle/phi/kernels/shape_kernel.cc
@@ -23,12 +23,9 @@ template <typename T, typename Context>
 void ShapeKernel(const Context& ctx,
                 const DenseTensor& input,
                 DenseTensor* out) {
-  auto in_var = &input;
+  auto& in_dims = input.dims();
-  phi::DDim in_dims;
+  out->Resize({in_dims.size()});
-  in_dims = in_var->dims();
+  auto out_data = ctx.template HostAlloc<int32_t>(out);
-  auto out_t = out;
-  out_t->Resize({in_dims.size()});
-  auto out_data = ctx.template HostAlloc<int32_t>(out_t);
  for (int i = 0; i < in_dims.size(); ++i) {
    out_data[i] = in_dims[i];
  }