fix set data type bug

test=develop

fix set data type bug
test=develop
565b19b7 · minqiyang · 8364688c · 565b19b7
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/imperative/layer.h paddle/fluid/imperative/layer.h +1 -1

未找到文件。
--- a/paddle/fluid/imperative/layer.h
+++ b/paddle/fluid/imperative/layer.h
@@ -195,7 +195,7 @@ class VarBase {
  // data type. e.g.. FP32
  inline void SetDataType(framework::proto::VarType::Type type) {
    auto tensor = var_->GetMutable<framework::LoDTensor>();
-    tensor->mutable_data(place_, type);
+    tensor->mutable_data(tensor->place(), type);
  }
  inline framework::proto::VarType::Type DataType() const {
    auto tensor = var_->Get<framework::LoDTensor>();