Fix pool2d_shape_check breakage as a result of API signature update.

[AshkanAliabadi]

Stack from ghstack:

• Force a sync on non-CPU tensors for the benchmark to reflect the timing accurately. #47714 Force a sync on non-CPU tensors for the benchmark to reflect the timing accurately.
• Tweak Vulkan memory use. #47728 Tweak Vulkan memory use.
• Update VMA. #47727 Update VMA.
• Fix Vulkan empty (and family) breakage as a result of API update. #47937 Fix Vulkan empty (and family) breakage as a result of API update.
• Fix pool2d_shape_check breakage as a result of API signature update. #47958 Fix pool2d_shape_check breakage as a result of API signature update.

[dr-ci]

💊 CI failures summary and remediations

As of commit bb4c4e7 (more details on the Dr. CI page):

---

💚 💚 Looks good so far! There are no failures yet. 💚 💚

---

This comment was automatically generated by Dr. CI (expand for details).

Follow this link to opt-out of these comments for your Pull Requests.

Please report bugs/suggestions on the GitHub issue tracker or post in the (internal) Dr. CI Users group.

See how this bot performed.

This comment has been revised 3 times.

[codecov]

Codecov Report

Merging #47958 (bb4c4e7) into gh/AshkanAliabadi/69/base (a9b6fa9) will decrease coverage by 0.00%.
The diff coverage is n/a.

@@                      Coverage Diff                      @@
##           gh/AshkanAliabadi/69/base   #47958      +/-   ##
=============================================================
- Coverage                      81.30%   81.30%   -0.01%     
=============================================================
  Files                           1839     1839              
  Lines                         198428   198428              
=============================================================
- Hits                          161326   161325       -1     
- Misses                         37102    37103       +1     

[github-actions]

Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as Stale.
Feel free to remove the Stale label if you feel this was a mistake.
Stale pull requests will automatically be closed 30 days after being marked Stale