[r2.5 port][ROCm] Port PR 47851 to r2.5

[deven-amd]

/cc @mihaimaruseac @angerson

#47851

[google-cla]

All (the pull request submitter and all commit authors) CLAs are signed, but one or more commits were authored or co-authored by someone other than the pull request submitter.

We need to confirm that all authors are ok with their commits being contributed to this project. Please have them confirm that by leaving a comment that contains only @googlebot I consent. in this pull request.

Note to project maintainer: There may be cases where the author cannot leave a comment, or the comment is not properly detected as consent. In those cases, you can manually confirm consent of the commit author(s), and set the cla label to yes (if enabled on your project).

ℹ️ Googlers: Go here for more info.

[gbaned]

@deven-amd Can you please sign CLA. Thanks!

[ekuznetsov139]

@googlebot I consent