ai-edge-torch-nightly 0.7.0.dev20251020__py3-none-any.whl → 0.7.0.dev20251021__py3-none-any.whl

ai_edge_torch/odml_torch/lowerings/_basic.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
+import logging
 import math
 import operator
 from typing import Optional, Union
@@ -101,9 +102,12 @@ def _hann_window_impl(
 def _aten_hann_window_default(
     lctx: LoweringContext,
     size: int,
-    *,
-    dtype: Optional[torch.dtype] = None,
+    **kwargs,
 ) -> ir.Value:
+  dtype = kwargs.pop("dtype", None)
+  layout = kwargs.pop("layout", torch.strided)
+  if layout != torch.strided:
+    logging.warning("hann_window only supports torch.strided layout.")
   return _hann_window_impl(lctx, size, True, dtype)
 
 
@@ -114,9 +118,12 @@ def _aten_hann_window_periodic(
     lctx: LoweringContext,
     size: int,
     periodic: bool,
-    *,
-    dtype: Optional[torch.dtype] = None,
+    **kwargs,
 ) -> ir.Value:
+  dtype = kwargs.pop("dtype", None)
+  layout = kwargs.pop("layout", torch.strided)
+  if layout != torch.strided:
+    logging.warning("hann_window only supports torch.strided layout.")
   return _hann_window_impl(lctx, size, periodic, dtype)
 
 

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.dev20251020"
+__version__ = "0.7.0.dev20251021"

{ai_edge_torch_nightly-0.7.0.dev20251020.dist-info → ai_edge_torch_nightly-0.7.0.dev20251021.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: ai-edge-torch-nightly
-Version: 0.7.0.dev20251020
+Version: 0.7.0.dev20251021
 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
@@ -37,17 +37,7 @@ Requires-Dist: ai-edge-quantizer-nightly
 Requires-Dist: jax
 Requires-Dist: torch-xla2[odml]>=0.0.1.dev20241201
 Provides-Extra: torch-xla
-Requires-Dist: torch_xla>=2.4.0; extra == "torch-xla"
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: keywords
-Dynamic: license-file
-Dynamic: provides-extra
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
+Requires-Dist: torch-xla>=2.4.0; extra == "torch-xla"
 
 Library that supports converting PyTorch models into a .tflite format, which can
 then be run with TensorFlow Lite and MediaPipe.  This enables applications for

{ai_edge_torch_nightly-0.7.0.dev20251020.dist-info → ai_edge_torch_nightly-0.7.0.dev20251021.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=7_nrKPXeG7PcqOTpJYoiFjOUpDJDKrRA6WU8ZIEVPDY,806
+ai_edge_torch/version.py,sha256=OqJtg7_I5JrjrCcK_c026Q_w0JSHTPq1q3hHLlxAzt8,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
@@ -250,7 +250,7 @@ ai_edge_torch/odml_torch/jax_bridge/__init__.py,sha256=e9Oa4J3An9FYr3zM0OzjzyNNi
 ai_edge_torch/odml_torch/jax_bridge/_wrap.py,sha256=LqwZ1vCJTSOzgzvH8LUAN-sAkF-l_pGj1AMEIzAqHCA,6638
 ai_edge_torch/odml_torch/jax_bridge/utils.py,sha256=T8isGc896VrHZ6c_L5pYmLpolQ7ibcOlgWfPuVFPzIg,2264
 ai_edge_torch/odml_torch/lowerings/__init__.py,sha256=uJ-niilt1c-D6QJzLwgvCUf62le_JsxQTlqj_iP_Ps0,1009
-ai_edge_torch/odml_torch/lowerings/_basic.py,sha256=-zKZtqvOx3COBhjCDtiZWMn5fY-boktkWXjp5Kepiro,14716
+ai_edge_torch/odml_torch/lowerings/_basic.py,sha256=kUbicoJcYZOgYRhydXYIsLyB2lW_Y-39skvjvrqhevo,15031
 ai_edge_torch/odml_torch/lowerings/_batch_norm.py,sha256=PaLI0BB6pdBW1VyfW8VTOT_Be-ZcqYdNOsyfzKfq8Cg,2064
 ai_edge_torch/odml_torch/lowerings/_convolution.py,sha256=Q0aDzyUcZMoSzSbOU-r3LJMgPe6fble0QwdYVIOHHHk,6887
 ai_edge_torch/odml_torch/lowerings/_decomp_registry.py,sha256=ybOdoFE5HIJTkyiYcc73zpyUyUpioVnAca6k0wyJPs4,2572
@@ -270,8 +270,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.dev20251020.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-ai_edge_torch_nightly-0.7.0.dev20251020.dist-info/METADATA,sha256=8b4KHY8SsXzytP_fKt5POyMHaxcoxbk5MOxToS4SPHo,2297
-ai_edge_torch_nightly-0.7.0.dev20251020.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-ai_edge_torch_nightly-0.7.0.dev20251020.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
-ai_edge_torch_nightly-0.7.0.dev20251020.dist-info/RECORD,,
+ai_edge_torch_nightly-0.7.0.dev20251021.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+ai_edge_torch_nightly-0.7.0.dev20251021.dist-info/METADATA,sha256=wB295znY-1RX_QXKMB8e-ujDBHQg_u39kqx8Pk8qleg,2074
+ai_edge_torch_nightly-0.7.0.dev20251021.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+ai_edge_torch_nightly-0.7.0.dev20251021.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
+ai_edge_torch_nightly-0.7.0.dev20251021.dist-info/RECORD,,

{ai_edge_torch_nightly-0.7.0.dev20251020.dist-info → ai_edge_torch_nightly-0.7.0.dev20251021.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (79.0.1)
+Generator: bdist_wheel (0.45.1)
 Root-Is-Purelib: true
 Tag: py3-none-any