updated expand.SymInt API in PT/XLA

Author: miladm

test/cpp/test_aten_xla_tensor.cpp

@@ -4655,7 +4655,7 @@ TEST_F(AtenXlaTensorTest, TestExpandSymInt) {
     torch::Tensor xla_y = torch::nonzero(xla_x);
     c10::SymInt xla_y0_size = xla_y.sym_sizes()[0];
     torch::Tensor xla_a = CopyToDevice(a, device);
-    torch::Tensor xla_b = xla_a.expand(
+    torch::Tensor xla_b = xla_a.expand_symint(
         c10::SymIntArrayRef({xla_y0_size, c10::SymInt(3), c10::SymInt(4)}),
         /*implicit=*/false);
     AllClose(b, xla_b);

torch_xla/csrc/aten_xla_type.cpp

@@ -1329,7 +1329,7 @@ at::Tensor XLANativeFunctions::expand(const at::Tensor& self,
       bridge::GetXlaTensor(self), torch::lazy::ToVector<int64_t>(size)));
 }
 
-at::Tensor XLANativeFunctions::expand(const at::Tensor& self,
+at::Tensor XLANativeFunctions::expand_symint(const at::Tensor& self,
                                       c10::SymIntArrayRef size, bool implicit) {
   XLA_FN_COUNTER("xla::");
   SymIntElements size_elements = SymIntElements(size);