fix infer shapes of pool_with_index (#40139)

* dbg pool infer shapes

* dbg

* fix format

paddle/fluid/operators/pool_with_index_op.cc

@@ -81,10 +81,14 @@ class MaxPoolWithIndexOp : public framework::OperatorWithKernel {
       output_shape.insert(output_shape.end(), ksize.begin(), ksize.end());
     } else {
       for (size_t i = 0; i < ksize.size(); ++i) {
+        if ((!ctx->IsRuntime()) && (in_x_dims[i + 2] < 0)) {
+          output_shape.push_back(in_x_dims[i + 2]);
+        } else {
           output_shape.push_back(MaxPoolOutputSize(in_x_dims[i + 2], ksize[i],
                                                    paddings[i], strides[i]));
         }
       }
+    }
     ctx->SetOutputDim("Out", phi::make_ddim(output_shape));
     ctx->SetOutputDim("Mask", phi::make_ddim(output_shape));
   }