BugFix with ParseInputDataType from LodTensorArray (#32918) (#32984)

* BugFix with ParseInputDataType from LodTensorArray * BugFix with ParseInputDataType from LodTensorArray
PaddlePaddle · May 20, 2021 · 8ecaa8a · 8ecaa8a
1 parent ef2ee5e
commit 8ecaa8a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/paddle/fluid/framework/operator.cc b/paddle/fluid/framework/operator.cc
@@ -1549,10 +1549,10 @@ void OperatorWithKernel::ParseInputDataType(
       } else if (var->IsType<SelectedRows>()) {
         t = &(var->Get<SelectedRows>().value());
       } else if (var->IsType<LoDTensorArray>()) {
-        auto t_arr = var->Get<LoDTensorArray>();
-        for (size_t j = 0; j < t_arr.size(); j++) {
-          if (t_arr[j].IsInitialized()) {
-            t = &(t_arr[j]);
+        auto t_arr = &var->Get<LoDTensorArray>();
+        for (size_t j = 0; j < t_arr->size(); j++) {
+          if (t_arr->at(j).IsInitialized()) {
+            t = &(t_arr->at(j));
           }
         }
       }