diff --git a/paddleslim/nas/ofa/layers.py b/paddleslim/nas/ofa/layers.py
index 18ca0c155c836fbb0562d4df27f4ea71e5fd8307..98763623a9b1cd639c1264f497e8cc180e5d403e 100644
--- a/paddleslim/nas/ofa/layers.py
+++ b/paddleslim/nas/ofa/layers.py
@@ -981,10 +981,10 @@ class SuperBatchNorm2D(nn.BatchNorm2D):
             batch_norm_out = core.ops.batch_norm(input, weight, bias, mean,
                                                  variance, mean_out_tmp,
                                                  variance_out_tmp, *attrs)
-            self._mean[:feature_dim] = mean
-            self._variance[:feature_dim] = variance
-            mean_out[:feature_dim] = mean_out_tmp
-            variance_out[:feature_dim] = variance_out_tmp
+            self._mean[:feature_dim].set_value(mean)
+            self._variance[:feature_dim].set_value(variance)
+            mean_out[:feature_dim].set_value(mean_out_tmp)
+            variance_out[:feature_dim].set_value(variance_out_tmp)
         else:
             batch_norm_out = core.ops.batch_norm(input, weight, bias,
                                                  self._mean, self._variance,
@@ -1031,10 +1031,10 @@ class SuperSyncBatchNorm(nn.SyncBatchNorm):
             sync_batch_norm_out, _, _, _, _, _ = core.ops.sync_batch_norm(
                 input, weight, bias, mean, variance, mean_out_tmp,
                 variance_out_tmp, *attrs)
-            self._mean[:feature_dim] = mean
-            self._variance[:feature_dim] = variance
-            mean_out[:feature_dim] = mean_out_tmp
-            variance_out[:feature_dim] = variance_out_tmp
+            self._mean[:feature_dim].set_value(mean)
+            self._variance[:feature_dim].set_value(variance)
+            mean_out[:feature_dim].set_value(mean_out_tmp)
+            variance_out[:feature_dim].set_value(variance_out_tmp)
         else:
             sync_batch_norm_out, _, _, _, _, _ = core.ops.sync_batch_norm(
                 input, weight, bias, self._mean, self._variance, mean_out,