Keep dynamic shape for tfl.fill ops by using SymInt.

PiperOrigin-RevId: 773058628

Author: junjiang-lab

ai_edge_torch/odml_torch/experimental/torch_tfl/_ops.py

@@ -130,8 +130,8 @@ def tfl_concatenation(
   return torch.cat(tensors, dim=dim)
 
 
-@custom_op_with_fake("tfl::fill", schema="(int[] x, Any y) -> Tensor")
-def tfl_fill(dims: Sequence[int], fill_value: Any) -> torch.Tensor:
+@custom_op_with_fake("tfl::fill", schema="(SymInt[] x, Any y) -> Tensor")
+def tfl_fill(dims: Sequence[torch.SymInt], fill_value: Any) -> torch.Tensor:
   return torch.full(dims, fill_value)