@@ -31,7 +31,7 @@ void SetLoD(DstLoD* dst, const SrcLoD& src) {
3131 }
3232}
3333
34- std::unique_ptr<pten::DenseTensor> MakePtenDenseTensorOrig (
34+ std::unique_ptr<pten::DenseTensor> MakePtenDenseTensorBase (
3535 const paddle::framework::Tensor& src) {
3636 VLOG (3 ) << " MakePtenDenseTensor based Tensor." ;
3737 pten::DenseTensorMeta meta{pten::TransToPtenDataType (src.type ()),
@@ -44,15 +44,15 @@ std::unique_ptr<pten::DenseTensor> MakePtenDenseTensorOrig(
4444}
4545
4646std::unique_ptr<pten::DenseTensor> MakePtenDenseTensor (
47- const paddle::framework::LoDTensor & src) {
48- auto out = MakePtenDenseTensorOrig (
47+ const paddle::framework::Tensor & src) {
48+ auto out = MakePtenDenseTensorBase (
4949 static_cast <const paddle::framework::Tensor&>(src));
5050 SetLoD (&(pten::CompatibleDenseTensorUtils::GetMutableMeta (out.get ())->lod ),
5151 src.lod ());
5252 return std::move (out);
5353}
5454
55- std::unique_ptr<pten::DenseTensor> MakePtenDenseTensorOrig (
55+ std::unique_ptr<pten::DenseTensor> MakePtenDenseTensorBase (
5656 const paddle::framework::Tensor& src, const pten::TensorArgDef& arg_def) {
5757 pten::DenseTensorMeta meta{
5858 arg_def.dtype , src.dims (), src.layout (), src.offset ()};
@@ -71,16 +71,15 @@ std::unique_ptr<pten::DenseTensor> MakePtenDenseTensorOrig(
7171}
7272
7373std::unique_ptr<pten::DenseTensor> MakePtenDenseTensor (
74- const paddle::framework::LoDTensor& src,
75- const pten::TensorArgDef& arg_def) {
76- auto out = MakePtenDenseTensorOrig (
74+ const paddle::framework::Tensor& src, const pten::TensorArgDef& arg_def) {
75+ auto out = MakePtenDenseTensorBase (
7776 static_cast <const paddle::framework::Tensor&>(src), arg_def);
7877 SetLoD (&(pten::CompatibleDenseTensorUtils::GetMutableMeta (out.get ())->lod ),
7978 src.lod ());
8079 return std::move (out);
8180}
8281
83- pten::Scalar MakePtenScalar (const paddle::framework::LoDTensor & src) {
82+ pten::Scalar MakePtenScalar (const paddle::framework::Tensor & src) {
8483 PADDLE_ENFORCE_EQ (src.numel (),
8584 1 ,
8685 paddle::platform::errors::InvalidArgument (
@@ -138,7 +137,7 @@ pten::Scalar MakePtenScalarFromVar(const framework::Variable& variable) {
138137 }
139138}
140139
141- pten::ScalarArray MakePtenScalarArray (const paddle::framework::LoDTensor & src) {
140+ pten::ScalarArray MakePtenScalarArray (const paddle::framework::Tensor & src) {
142141 if (src.type () == paddle::framework::proto::VarType::INT64) {
143142 return {src.data <int64_t >(), src.numel ()};
144143 } else if (src.type () == paddle::framework::proto::VarType::INT32) {
@@ -295,7 +294,7 @@ std::unique_ptr<pten::TensorBase> MakePtenTensorBaseFromVar(
295294 return {};
296295}
297296
298- void MovesStorageOrig (pten::DenseTensor* src, paddle::framework::Tensor* dst) {
297+ void MovesStorageBase (pten::DenseTensor* src, paddle::framework::Tensor* dst) {
299298 PADDLE_ENFORCE_NOT_NULL (
300299 src,
301300 platform::errors::InvalidArgument (
@@ -313,12 +312,12 @@ void MovesStorageOrig(pten::DenseTensor* src, paddle::framework::Tensor* dst) {
313312 dst->set_offset (src->meta ().offset );
314313}
315314
316- void MovesStorage (pten::DenseTensor* src, paddle::framework::LoDTensor * dst) {
317- MovesStorageOrig (src, static_cast <paddle::framework::Tensor*>(dst));
315+ void MovesStorage (pten::DenseTensor* src, paddle::framework::Tensor * dst) {
316+ MovesStorageBase (src, static_cast <paddle::framework::Tensor*>(dst));
318317 SetLoD (dst->mutable_lod (), src->lod ());
319318}
320319
321- void SharesStorageOrig (pten::DenseTensor* src, paddle::framework::Tensor* dst) {
320+ void SharesStorageBase (pten::DenseTensor* src, paddle::framework::Tensor* dst) {
322321 PADDLE_ENFORCE_NOT_NULL (
323322 src,
324323 platform::errors::InvalidArgument (
@@ -335,12 +334,12 @@ void SharesStorageOrig(pten::DenseTensor* src, paddle::framework::Tensor* dst) {
335334 dst->set_offset (src->meta ().offset );
336335}
337336
338- void SharesStorage (pten::DenseTensor* src, paddle::framework::LoDTensor * dst) {
339- SharesStorageOrig (src, static_cast <paddle::framework::Tensor*>(dst));
337+ void SharesStorage (pten::DenseTensor* src, paddle::framework::Tensor * dst) {
338+ SharesStorageBase (src, static_cast <paddle::framework::Tensor*>(dst));
340339 SetLoD (dst->mutable_lod (), src->lod ());
341340}
342341
343- void ReMakePtenDenseTensorOrig (const paddle::framework::Tensor& src,
342+ void ReMakePtenDenseTensorBase (const paddle::framework::Tensor& src,
344343 pten::DenseTensor* dst) {
345344 VLOG (3 ) << " ReMakePtenDenseTensor based Tensor." ;
346345 auto * meta = pten::CompatibleDenseTensorUtils::GetMutableMeta (dst);
@@ -363,15 +362,15 @@ void ReMakePtenDenseTensorOrig(const paddle::framework::Tensor& src,
363362 shared_storage->ResetAllocation (src.Holder ());
364363}
365364
366- void ReMakePtenDenseTensor (const paddle::framework::LoDTensor & src,
365+ void ReMakePtenDenseTensor (const paddle::framework::Tensor & src,
367366 pten::DenseTensor* dst) {
368367 auto * meta = pten::CompatibleDenseTensorUtils::GetMutableMeta (dst);
369368 SetLoD (&meta->lod , src.lod ());
370- ReMakePtenDenseTensorOrig (static_cast <const paddle::framework::Tensor&>(src),
369+ ReMakePtenDenseTensorBase (static_cast <const paddle::framework::Tensor&>(src),
371370 dst);
372371}
373372
374- void ReMakePtenDenseTensorByArgDefOrig (const paddle::framework::Tensor& src,
373+ void ReMakePtenDenseTensorByArgDefBase (const paddle::framework::Tensor& src,
375374 const pten::TensorArgDef& arg_def,
376375 pten::DenseTensor* dst) {
377376 VLOG (3 ) << " ReMakePtenDenseTensor based Tensor and TensorArgDef." ;
@@ -397,12 +396,12 @@ void ReMakePtenDenseTensorByArgDefOrig(const paddle::framework::Tensor& src,
397396 }
398397}
399398
400- void ReMakePtenDenseTensorByArgDef (const paddle::framework::LoDTensor & src,
399+ void ReMakePtenDenseTensorByArgDef (const paddle::framework::Tensor & src,
401400 const pten::TensorArgDef& arg_def,
402401 pten::DenseTensor* dst) {
403402 auto * meta = pten::CompatibleDenseTensorUtils::GetMutableMeta (dst);
404403 SetLoD (&meta->lod , src.lod ());
405- ReMakePtenDenseTensorByArgDefOrig (
404+ ReMakePtenDenseTensorByArgDefBase (
406405 static_cast <const paddle::framework::Tensor&>(src), arg_def, dst);
407406}
408407
0 commit comments