tf-models-nightly 2.20.0.dev20250617__py2.py3-none-any.whl → 2.20.0.dev20250619__py2.py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of tf-models-nightly might be problematic. Click here for more details.

Files changed (7) hide show
  1. orbit/utils/loop_fns.py +2 -2
  2. {tf_models_nightly-2.20.0.dev20250617.dist-info → tf_models_nightly-2.20.0.dev20250619.dist-info}/METADATA +1 -1
  3. {tf_models_nightly-2.20.0.dev20250617.dist-info → tf_models_nightly-2.20.0.dev20250619.dist-info}/RECORD +7 -7
  4. {tf_models_nightly-2.20.0.dev20250617.dist-info → tf_models_nightly-2.20.0.dev20250619.dist-info}/AUTHORS +0 -0
  5. {tf_models_nightly-2.20.0.dev20250617.dist-info → tf_models_nightly-2.20.0.dev20250619.dist-info}/LICENSE +0 -0
  6. {tf_models_nightly-2.20.0.dev20250617.dist-info → tf_models_nightly-2.20.0.dev20250619.dist-info}/WHEEL +0 -0
  7. {tf_models_nightly-2.20.0.dev20250617.dist-info → tf_models_nightly-2.20.0.dev20250619.dist-info}/top_level.txt +0 -0
orbit/utils/loop_fns.py CHANGED
@@ -20,7 +20,7 @@ from orbit.utils import tpu_summaries
20
20
  import tensorflow as tf, tf_keras
21
21
 
22
22
  # pylint: disable=g-direct-tensorflow-import
23
- from tensorflow.python.tpu.google.sparse_core import tpu_embedding_v3
23
+ from tensorflow.python.tpu import embedding_context_utils as ecu
24
24
  # pylint: enable=g-direct-tensorflow-import
25
25
 
26
26
 
@@ -204,7 +204,7 @@ class LoopFnWithSummaries(tpu_summaries.OptionalSummariesFunction):
204
204
 
205
205
  def __call__(self, iterator, num_steps):
206
206
  if tf.summary.should_record_summaries():
207
- with tpu_embedding_v3.SequentialEmbeddingContext():
207
+ with ecu.SequentialEmbeddingContext():
208
208
  output = self.with_summaries(iterator, tf.constant(1))
209
209
  num_steps -= 1
210
210
  if num_steps >= 1:
{tf_models_nightly-2.20.0.dev20250617.dist-info → tf_models_nightly-2.20.0.dev20250619.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: tf-models-nightly
3
- Version: 2.20.0.dev20250617
3
+ Version: 2.20.0.dev20250619
4
4
  Summary: TensorFlow Official Models
5
5
  Home-page: https://github.com/tensorflow/models
6
6
  Author: Google Inc.
{tf_models_nightly-2.20.0.dev20250617.dist-info → tf_models_nightly-2.20.0.dev20250619.dist-info}/RECORD RENAMED
@@ -1238,7 +1238,7 @@ orbit/utils/__init__.py,sha256=I34xJSdKQtpdsWS91RoVRdCXgLzRQy6sY0Ijkka3czU,1218
1238
1238
  orbit/utils/common.py,sha256=HV50Iy2XnYJ7qSnAk7MohQgNQ5TnLQv8gJIHHX2qb9U,3923
1239
1239
  orbit/utils/common_test.py,sha256=4I643Z04lXqu1E60a2FwbZxtChZ66jtBHowkdaFkSgY,1042
1240
1240
  orbit/utils/epoch_helper.py,sha256=k4cuJQXhzKd5hOtgR_jB-vbrAjNFlFg-w0D4-kz8IYw,2146
1241
- orbit/utils/loop_fns.py,sha256=wlmD6x6UIM3rwaxhpG7oNLDs9yBbcoG3SXyYsQo9pts,8246
1241
+ orbit/utils/loop_fns.py,sha256=mbGadLkbkiTgnsL8a_Pz061j8dczBMvMEZXOAplQ-Sc,8228
1242
1242
  orbit/utils/summary_manager.py,sha256=_NykIGSl4wLeqzYDjbJTf5RAfn9Jweckr65aHzvUGd0,4287
1243
1243
  orbit/utils/summary_manager_interface.py,sha256=yTX8wt1EjcqRGNxabkYeN2nL_Zk10XExlRp9WoWDp6M,2253
1244
1244
  orbit/utils/tpu_summaries.py,sha256=-BD6LLW-V_qxQWmAMjtu76mrngmCpwtd6Wf--k7mVaA,5641
@@ -1248,9 +1248,9 @@ tensorflow_models/tensorflow_models_test.py,sha256=yiAneltAW3NHSj3fUSvHNBjfq0MGZ
1248
1248
  tensorflow_models/nlp/__init__.py,sha256=8uQd4wI6Zc4IJMPjtQifMeWVbPFkTxqYh66wfivCOL4,807
1249
1249
  tensorflow_models/uplift/__init__.py,sha256=NzaweFf4ZmhRb2l_fuV6bP-2N8oSO3xu6xJqVb1UmpY,999
1250
1250
  tensorflow_models/vision/__init__.py,sha256=ks420Ooqzi0hU7HnQpM5rylLaE-YcJdJkBx_umVaXlE,833
1251
- tf_models_nightly-2.20.0.dev20250617.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
1252
- tf_models_nightly-2.20.0.dev20250617.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
1253
- tf_models_nightly-2.20.0.dev20250617.dist-info/METADATA,sha256=2qe29nvqB3hjQ_s67dCOCSFtLbysyYv1iYQMfNe7yBA,1432
1254
- tf_models_nightly-2.20.0.dev20250617.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
1255
- tf_models_nightly-2.20.0.dev20250617.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
1256
- tf_models_nightly-2.20.0.dev20250617.dist-info/RECORD,,
1251
+ tf_models_nightly-2.20.0.dev20250619.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
1252
+ tf_models_nightly-2.20.0.dev20250619.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
1253
+ tf_models_nightly-2.20.0.dev20250619.dist-info/METADATA,sha256=m_5vbKlwzO5LIH_U96rXCEAXNzpwtHNULJW0QgnL6kY,1432
1254
+ tf_models_nightly-2.20.0.dev20250619.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
1255
+ tf_models_nightly-2.20.0.dev20250619.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
1256
+ tf_models_nightly-2.20.0.dev20250619.dist-info/RECORD,,