File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
paddle/pten/api/lib/utils Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -292,7 +292,7 @@ void MakeVariableFromPtenTensor(pten::DenseTensor* src,
292292 if (variable->IsType <framework::LoDTensor>()) {
293293 auto * tensor = variable->GetMutable <framework::LoDTensor>();
294294
295- auto dtype = pten::TransToProtoVarType (src->data_type ());
295+ auto dtype = pten::TransToProtoVarType (src->dtype ());
296296 tensor->Resize (src->dims ());
297297 SetLoD (tensor->mutable_lod (), src->lod ());
298298
@@ -314,7 +314,7 @@ void MakeVariableFromPtenTensor(pten::DenseTensor* src,
314314
315315 } else if (variable->IsType <framework::SelectedRows>()) {
316316 auto * tensor = variable->GetMutable <framework::SelectedRows>();
317- auto dtype = pten::TransToProtoVarType (src->data_type ());
317+ auto dtype = pten::TransToProtoVarType (src->dtype ());
318318
319319 if (tensor->value ().IsInitialized ()) {
320320 } else {
You can’t perform that action at this time.
0 commit comments