readd share tensor impl methods

YuanRisheng · Oct 20, 2021 · ab8db2d · ab8db2d
1 parent 04cf058
commit ab8db2d
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/paddle/fluid/framework/pten_utils.cc b/paddle/fluid/framework/pten_utils.cc
@@ -79,6 +79,20 @@ std::shared_ptr<pten::DenseTensor> MakeTensorImpl<pten::DenseTensor>(
       pten::TransToPtDataLayout(tensor.layout()));
 }
 
+template <>
+void ShareTensorImpl<pten::DenseTensor>(pten::DenseTensor* tensor_impl,
+                                        LoDTensor* out) {
+  out->ResetHolderWithType(tensor_impl->allocation(),
+                           pten::TransToProtoVarType(tensor_impl->data_type()));
+}
+
+template <>
+void ShareTensorImpl<pten::DenseTensor>(pten::DenseTensor* tensor_impl,
+                                        Tensor* out) {
+  out->ResetHolderWithType(tensor_impl->allocation(),
+                           pten::TransToProtoVarType(tensor_impl->data_type()));
+}
+
 std::shared_ptr<pten::TensorBase> InputVariableToPtTensor(
     const framework::Variable& variable, const pten::TensorArgDef& arg_def) {
   auto expected_place = pten::TransToFluidPlace(arg_def.backend);