diff --git a/paddle/fluid/framework/new_exec.h b/paddle/fluid/framework/new_exec.h
index 1ce87ecc938816bbffc35501cc9e874557ddb5b4..2c4ff7588eedf95bd60a15f520b2e1b8ee490fee 100644
--- a/paddle/fluid/framework/new_exec.h
+++ b/paddle/fluid/framework/new_exec.h
@@ -86,9 +86,9 @@ class ExecutionContextV2 : public ExecutionContext {
 
   std::vector<std::string> InNameList() const {
     std::vector<std::string> vec_temp;
-    vec_temp.reserve(ctx_.output_name_map.size());
+    vec_temp.reserve(ctx_.input_name_map.size());
 
-    for (auto& input : ctx_.output_name_map) {
+    for (auto& input : ctx_.input_name_map) {
       vec_temp.push_back(input.first);
     }