tf-models-nightly 2.20.0.dev20250612__py2.py3-none-any.whl → 2.20.0.dev20250614__py2.py3-none-any.whl

orbit/utils/loop_fns.py

@@ -19,6 +19,10 @@ from orbit.utils import tpu_summaries
 
 import tensorflow as tf, tf_keras
 
+# pylint: disable=g-direct-tensorflow-import
+from tensorflow.python.tpu.google.sparse_core import tpu_embedding_v3
+# pylint: enable=g-direct-tensorflow-import
+
 
 def create_loop_fn(step_fn):
   """Creates a loop function driven by a Python `while` loop.
@@ -200,7 +204,8 @@ class LoopFnWithSummaries(tpu_summaries.OptionalSummariesFunction):
 
   def __call__(self, iterator, num_steps):
     if tf.summary.should_record_summaries():
-      output = self.with_summaries(iterator, tf.constant(1))
+      with tpu_embedding_v3.SequentialEmbeddingContext():
+        output = self.with_summaries(iterator, tf.constant(1))
       num_steps -= 1
     if num_steps >= 1:
       output = self.without_summaries(iterator, num_steps)

{tf_models_nightly-2.20.0.dev20250612.dist-info → tf_models_nightly-2.20.0.dev20250614.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tf-models-nightly
-Version: 2.20.0.dev20250612
+Version: 2.20.0.dev20250614
 Summary: TensorFlow Official Models
 Home-page: https://github.com/tensorflow/models
 Author: Google Inc.

{tf_models_nightly-2.20.0.dev20250612.dist-info → tf_models_nightly-2.20.0.dev20250614.dist-info}/RECORD

@@ -1238,7 +1238,7 @@ orbit/utils/__init__.py,sha256=I34xJSdKQtpdsWS91RoVRdCXgLzRQy6sY0Ijkka3czU,1218
 orbit/utils/common.py,sha256=HV50Iy2XnYJ7qSnAk7MohQgNQ5TnLQv8gJIHHX2qb9U,3923
 orbit/utils/common_test.py,sha256=4I643Z04lXqu1E60a2FwbZxtChZ66jtBHowkdaFkSgY,1042
 orbit/utils/epoch_helper.py,sha256=k4cuJQXhzKd5hOtgR_jB-vbrAjNFlFg-w0D4-kz8IYw,2146
-orbit/utils/loop_fns.py,sha256=zjgN1FiiGU53J-eqTIfB2CgxA0VZ8b5dAfFwBXnGeDA,8026
+orbit/utils/loop_fns.py,sha256=wlmD6x6UIM3rwaxhpG7oNLDs9yBbcoG3SXyYsQo9pts,8246
 orbit/utils/summary_manager.py,sha256=_NykIGSl4wLeqzYDjbJTf5RAfn9Jweckr65aHzvUGd0,4287
 orbit/utils/summary_manager_interface.py,sha256=yTX8wt1EjcqRGNxabkYeN2nL_Zk10XExlRp9WoWDp6M,2253
 orbit/utils/tpu_summaries.py,sha256=-BD6LLW-V_qxQWmAMjtu76mrngmCpwtd6Wf--k7mVaA,5641
@@ -1248,9 +1248,9 @@ tensorflow_models/tensorflow_models_test.py,sha256=yiAneltAW3NHSj3fUSvHNBjfq0MGZ
 tensorflow_models/nlp/__init__.py,sha256=8uQd4wI6Zc4IJMPjtQifMeWVbPFkTxqYh66wfivCOL4,807
 tensorflow_models/uplift/__init__.py,sha256=NzaweFf4ZmhRb2l_fuV6bP-2N8oSO3xu6xJqVb1UmpY,999
 tensorflow_models/vision/__init__.py,sha256=ks420Ooqzi0hU7HnQpM5rylLaE-YcJdJkBx_umVaXlE,833
-tf_models_nightly-2.20.0.dev20250612.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.20.0.dev20250612.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.20.0.dev20250612.dist-info/METADATA,sha256=UgInAY9LGCEYLFpOnb70t0Ij0HqAEx3ZI9__UFQ5lLY,1432
-tf_models_nightly-2.20.0.dev20250612.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.20.0.dev20250612.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.20.0.dev20250612.dist-info/RECORD,,
+tf_models_nightly-2.20.0.dev20250614.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.20.0.dev20250614.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.20.0.dev20250614.dist-info/METADATA,sha256=rApeF7pnw59UkrZ2F_O1OBJ0QFaORjiOZwkcciJVtO0,1432
+tf_models_nightly-2.20.0.dev20250614.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.20.0.dev20250614.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.20.0.dev20250614.dist-info/RECORD,,