Do not copy beyond out tensor bounds

[pytorchbot]

Summary: set_all_to_value(out_data, out_step_len, value) loops over the padding and copies a row at a time.

Add some checks to make sure we do not exceed the out tensor (out_data) size when copying.

Differential Revision: D86436435

[pytorch-bot[bot]]

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/15717

• :page_facing_up: Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

:x: 1 New Failure, 3 Cancelled Jobs

As of commit 9567f13cb0d15823ff4fe353095c70205b1a837b with merge base e0dda9059d082537cee36be6c5e4fe3b18c880c0 ():

NEW FAILURE - The following job has failed:

• pull / test-moshi-linux / linux-job (gh) RuntimeError: Command docker exec -t 08a9394b8e77e943c7fc37d7ee2f5b4dbac595c4a177b1f2b68381bdad314a81 /exec failed with exit code 1

CANCELLED JOBS - The following jobs were cancelled. Please retry:

• pull / test-llava-runner-linux / linux-job (gh) ##[error]The operation was canceled.
• pull / unittest / macos / macos-job (gh) ##[error]The operation was canceled.
• pull / unittest-editable / macos / macos-job (gh) ##[error]The operation was canceled.

This comment was automatically generated by Dr. CI and updates every 15 minutes.