ai-edge-torch-nightly 0.7.0.dev20250911__py3-none-any.whl → 0.7.0.dev20250912__py3-none-any.whl

ai_edge_torch/_convert/test/test_convert.py

@@ -636,6 +636,50 @@ class TestConvert(googletest.TestCase):
       self.fail(f"Conversion failed with 6d inputs: {err}")
     # pylint: enable=broad-except
 
+  @googletest.skipIf(
+      ai_edge_torch.config.in_oss,
+      reason="wait until dependencies are released to tf-nightly",
+  )
+  def test_convert_model_with_slice_6d_inputs(self):
+    """Test converting a simple model with slice and 6d inputs."""
+
+    class SampleModel(nn.Module):
+
+      def forward(self, x: torch.Tensor):
+        return x[0:1, 0:2, 0:3, 0:4, 0:5, 0:1]
+
+    model = SampleModel().eval()
+    args = (torch.randn((1, 2, 3, 4, 5, 6)),)
+
+    try:
+      # Expect this to fix the error during conversion
+      ai_edge_torch.convert(model, args)
+    except Exception as err:
+      self.fail(f"Conversion failed with 6d inputs for slice: {err}")
+    # pylint: enable=broad-except
+
+  @googletest.skipIf(
+      ai_edge_torch.config.in_oss,
+      reason="wait until dependencies are released to tf-nightly",
+  )
+  def test_convert_model_with_strided_slice_6d_inputs(self):
+    """Test converting a simple model with strided_slice and 6d inputs."""
+
+    class SampleModel(nn.Module):
+
+      def forward(self, x: torch.Tensor):
+        return x[:, :, :, :, :, ::2]
+
+    model = SampleModel().eval()
+    args = (torch.randn((1, 2, 3, 4, 5, 6)),)
+
+    try:
+      # Expect this to fix the error during conversion
+      ai_edge_torch.convert(model, args)
+    except Exception as err:
+      self.fail(f"Conversion failed with 6d inputs for strided_slice: {err}")
+    # pylint: enable=broad-except
+
   def test_compile_model(self):
     """Tests AOT compilation of a simple Add module."""
 

ai_edge_torch/version.py

@@ -15,4 +15,4 @@
 
 # The next version of ai-edge-torch.
 # The minor version code should be bumped after every release.
-__version__ = "0.7.0.dev20250911"
+__version__ = "0.7.0.dev20250912"

{ai_edge_torch_nightly-0.7.0.dev20250911.dist-info → ai_edge_torch_nightly-0.7.0.dev20250912.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ai-edge-torch-nightly
-Version: 0.7.0.dev20250911
+Version: 0.7.0.dev20250912
 Summary: Supporting PyTorch models with the Google AI Edge TFLite runtime.
 Home-page: https://github.com/google-ai-edge/ai-edge-torch
 Keywords: On-Device ML,AI,Google,TFLite,PyTorch,LLMs,GenAI

{ai_edge_torch_nightly-0.7.0.dev20250911.dist-info → ai_edge_torch_nightly-0.7.0.dev20250912.dist-info}/RECORD

@@ -2,7 +2,7 @@ ai_edge_torch/__init__.py,sha256=lemyLCNoGYRnJsmDuGZu7qOqLbLqG6CGDFtu3ue1syU,129
 ai_edge_torch/_config.py,sha256=AiqhbcheF7j_ozIGDLC89k1we95aVgFDa-tR6h7UI0s,2529
 ai_edge_torch/conftest.py,sha256=r0GTrhMRhlmOGrrkvumHN8hkmyug6WvF60vWq8wRIBI,758
 ai_edge_torch/model.py,sha256=A7loFu8jE9CsXsfMmHYZ-KDFJiaD8Kkqwm_9d3IVzk0,5638
-ai_edge_torch/version.py,sha256=ad9ZkVC0B8CG30qp_HL1btX4L7zZ8wC0T5_b6bbtVOQ,806
+ai_edge_torch/version.py,sha256=DHypu0sPpwondGn5ZEll2UHAaf_4N5r5pHQU9FA9Y7s,806
 ai_edge_torch/_convert/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/_convert/conversion.py,sha256=iQk3R-pLq4c1nfLqPB4xTRj78gghxPGzJCJtILLdg5o,6123
 ai_edge_torch/_convert/conversion_utils.py,sha256=Sr8qXVcTwc-ZnZmK7yxVrIOOp1S_vNrwzC0zUvLTI2o,2160
@@ -27,7 +27,7 @@ ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_partitio
 ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_partitioners/greedy.py,sha256=L_x8BrF7UDah-SYl-pG11I6CIckdU9kBTUHcmwW4cts,2420
 ai_edge_torch/_convert/fx_passes/optimize_layout_transposes_pass/layout_partitioners/min_cut.py,sha256=oXbr9G5Jc21xd1dr2CDrp774I4crs0_kkN490K5fNn0,7312
 ai_edge_torch/_convert/test/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
-ai_edge_torch/_convert/test/test_convert.py,sha256=m0j4pLcVYEhLsO47z8K9wnE-7fjAmhkZfjtXHXBLbGo,20591
+ai_edge_torch/_convert/test/test_convert.py,sha256=rZZg04rY3Q0w_qv8x4vwBkedwRd0MwgsacSX2-AN688,21991
 ai_edge_torch/_convert/test/test_convert_composites.py,sha256=BCIODgxMI_3MxMLfNWYMGjcz-al-J3z5eDHCiZJXNwY,7992
 ai_edge_torch/_convert/test/test_convert_multisig.py,sha256=6_C2R9--KyNR7_oezZIAfyTSR97tOeEWy4XGcbSxBDE,5778
 ai_edge_torch/_convert/test/test_to_channel_last_io.py,sha256=1o-gUiwzIuO67FNAJ8DeyKv8fVUeZVNNNwofNVDjYeU,3024
@@ -269,8 +269,8 @@ ai_edge_torch/testing/__init__.py,sha256=_yGgvnBZWb7T3IN3mc4x1sS4vM96HZwM8pwIcPG
 ai_edge_torch/testing/export.py,sha256=k5mGDGzwc23Z4zaIVDs8CNh-oOt64gsf9MS9NjhbPy4,3293
 ai_edge_torch/testing/model_coverage/__init__.py,sha256=5P8J6Zk5YYtDvTBucFvB9NGSRI7Gw_24WnrbhXgycEE,765
 ai_edge_torch/testing/model_coverage/model_coverage.py,sha256=UPB448aMDUyC0HNYVqio2rcJPnDN0tBQMP08J6vPYew,4718
-ai_edge_torch_nightly-0.7.0.dev20250911.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-ai_edge_torch_nightly-0.7.0.dev20250911.dist-info/METADATA,sha256=YwLdah0CL4_mA0ZKe9npxWkCI5S6MFcokhGces5SCvQ,2074
-ai_edge_torch_nightly-0.7.0.dev20250911.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-ai_edge_torch_nightly-0.7.0.dev20250911.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
-ai_edge_torch_nightly-0.7.0.dev20250911.dist-info/RECORD,,
+ai_edge_torch_nightly-0.7.0.dev20250912.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+ai_edge_torch_nightly-0.7.0.dev20250912.dist-info/METADATA,sha256=ni2ExO5_iIpKdawP63apxP61T6fmph4TheU7x3WNCfI,2074
+ai_edge_torch_nightly-0.7.0.dev20250912.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+ai_edge_torch_nightly-0.7.0.dev20250912.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
+ai_edge_torch_nightly-0.7.0.dev20250912.dist-info/RECORD,,