Add repo-name param for calculate-docker-image action (#7205)

To extend the calculate-docker-image action used in other repo, such as
pytorch/ao.

Refer
https://github.com/pytorch/ao/actions/runs/17938785568/job/51010166034#step:14:85,
we're enabling xpu test in torchao and want to leverage pytorch ci image
to do the test.

Author: chuanqi129

.github/actions/calculate-docker-image/action.yml

@@ -42,6 +42,9 @@ inputs:
     description: If set to true, always push the image to ECR even if it exists.
     required: false
     default: false
+  repo-name:
+    description: The name of the repository containing the docker-build-dir.
+    required: false
 
 outputs:
   docker-image:
@@ -56,7 +59,7 @@ runs:
       shell: bash
       working-directory: ${{ inputs.working-directory }}
       env:
-        REPO_NAME: ${{ github.event.repository.name }}
+        REPO_NAME: ${{ inputs.repo-name || github.event.repository.name }}
         DOCKER_IMAGE_NAME: ${{ inputs.docker-image-name }}
         DOCKER_BUILD_DIR: ${{ inputs.docker-build-dir }}
         DOCKER_BUILD_SCRIPT: ${{ inputs.docker-build-script }}