Fix py_func_op's problem

paddle/fluid/operators/controlflow/get_places_op.cc

@@ -93,9 +93,9 @@ execution.
 
 class GetPlacesInferVarType : public framework::VarTypeInference {
  public:
-  void operator()(framework::InferVarTypeContext &ctx) const override {
+  void operator()(framework::InferVarTypeContext *ctx) const override {
-    for (auto &o_name : ctx.Output("Out")) {
+    for (auto &o_name : ctx->Output("Out")) {
-      ctx.SetType(o_name, framework::proto::VarType::PLACE_LIST);
+      ctx->SetType(o_name, framework::proto::VarType::PLACE_LIST);
     }
   }
 };

paddle/fluid/operators/py_func_op.cc

@@ -99,7 +99,7 @@ class PyFuncOpVarTypeInference : public framework::VarTypeInference {
   void operator()(framework::InferVarTypeContext *ctx) const override {
     bool has_out = (ctx->HasOutput("Out") && !ctx->Output("Out").empty());
 
-    bool has_in = (ctx->HasInput("X") && !ctx->Input("Out").empty());
+    bool has_in = (ctx->HasInput("X") && !ctx->Input("X").empty());
 
     /**
      * X or Out can be empty, so that py_func can be more flexible