"fix style" (#13094)

76b0a384 · dzhwinter · dongzhihong · cfe611f1 · 76b0a384
显示空白变更内容
内联并排

Showing with 4 addition and 1 deletion

paddle/fluid/operators/shrink_rnn_memory_op.cc paddle/fluid/operators/shrink_rnn_memory_op.cc +4 -1

未找到文件。
--- a/paddle/fluid/operators/shrink_rnn_memory_op.cc
+++ b/paddle/fluid/operators/shrink_rnn_memory_op.cc
@@ -62,7 +62,10 @@ class ShrinkRNNMemoryOp : public ArrayOp {
    }

    if (dst_num_rows != 0) {
-      out_tensor.ShareDataWith(x_tensor.Slice(0, height));
+      out_tensor.mutable_data(place, x_tensor.type());
+      auto dev_ctx = platform::DeviceContextPool::Instance().Get(place);
+      framework::TensorCopy(x_tensor.Slice(0, height), place, *dev_ctx,
+                            &out_tensor);
    }
  }
 };