diff --git a/paddle/fluid/framework/ir/mkldnn/operator_scale_onednn_fuse_pass.cc b/paddle/fluid/framework/ir/mkldnn/operator_scale_onednn_fuse_pass.cc
index 2759c79b7a7d93526e2b90342d82191e71c25c83..2e4163feb8b4ab44f868b630a2631bbcad88a7c1 100644
--- a/paddle/fluid/framework/ir/mkldnn/operator_scale_onednn_fuse_pass.cc
+++ b/paddle/fluid/framework/ir/mkldnn/operator_scale_onednn_fuse_pass.cc
@@ -73,7 +73,7 @@ void FuseOperatorScaleOneDNNPass::FuseScale(Graph *graph,
       auto *scale_var = scope->FindVar(scale_var_name);
       // ScaleTensor must be weight
       if (scale_var == nullptr) return;
-      auto *scale_tensor = scale_var->GetMutable<LoDTensor>();
+      auto *scale_tensor = scale_var->GetMutable<phi::DenseTensor>();
       scale = *(scale_tensor->data<float>());
     }
 
diff --git a/paddle/fluid/inference/tensorrt/convert/preln_layernorm_shift_partition_op.cc b/paddle/fluid/inference/tensorrt/convert/preln_layernorm_shift_partition_op.cc
index 7b7719e3ac819bc51b507c0e4da7a858c8eee00a..402cd363aba98850d2f2ae8d9dd23461f40cd954 100644
--- a/paddle/fluid/inference/tensorrt/convert/preln_layernorm_shift_partition_op.cc
+++ b/paddle/fluid/inference/tensorrt/convert/preln_layernorm_shift_partition_op.cc
@@ -49,8 +49,8 @@ class PrelnLayerNormShiftPartitionOpConverter : public OpConverter {
             ? PADDLE_GET_CONST(int, op_desc.GetAttr("shift_size"))
             : 0;
 
-    auto* Bias_t = Bias_v->GetMutable<framework::LoDTensor>();
-    auto* Scale_t = Scale_v->GetMutable<framework::LoDTensor>();
+    auto* Bias_t = Bias_v->GetMutable<phi::DenseTensor>();
+    auto* Scale_t = Scale_v->GetMutable<phi::DenseTensor>();
 
     auto bias_weight =
         engine_->GetFp32TrtWeight(op_desc.Input("Bias").front(), *Bias_t);