Rename input

pytorch · Nov 30, 2023 · 48e9943 · 48e9943
1 parent 99e697d
commit 48e9943
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_xla/csrc/init_python_bindings.cpp b/torch_xla/csrc/init_python_bindings.cpp
@@ -1816,10 +1816,10 @@ void InitXlaModuleBindings(py::module m) {
   // of the shards returned from `_get_local_shards`.
   m.def(
       "_get_local_shard_replica_and_indices",
-      [](const std::vector<at::Tensor>& input)
+      [](const std::vector<at::Tensor>& input_tensors)
           -> std::vector<std::vector<std::pair<int, py::object>>> {
         std::vector<std::vector<std::pair<int, py::object>>> result;
-        for (auto& tensor : input) {
+        for (auto& tensor : input_tensors) {
           XLATensorPtr xtensor = bridge::GetXlaTensor(tensor);
           XLA_CHECK(xtensor->sharding_spec() != nullptr)
               << "Tensor is not sharded";