tf-keras-nightly 2.19.0.dev2024101609__py3-none-any.whl → 2.19.0.dev2024101809__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.
- tf_keras/__init__.py +1 -1
- tf_keras/src/engine/node.py +8 -3
- {tf_keras_nightly-2.19.0.dev2024101609.dist-info → tf_keras_nightly-2.19.0.dev2024101809.dist-info}/METADATA +1 -1
- {tf_keras_nightly-2.19.0.dev2024101609.dist-info → tf_keras_nightly-2.19.0.dev2024101809.dist-info}/RECORD +6 -6
- {tf_keras_nightly-2.19.0.dev2024101609.dist-info → tf_keras_nightly-2.19.0.dev2024101809.dist-info}/WHEEL +1 -1
- {tf_keras_nightly-2.19.0.dev2024101609.dist-info → tf_keras_nightly-2.19.0.dev2024101809.dist-info}/top_level.txt +0 -0
tf_keras/__init__.py
CHANGED
tf_keras/src/engine/node.py
CHANGED
@@ -84,9 +84,10 @@ class Node:
|
|
84
84
|
self.call_args = call_args
|
85
85
|
self.call_kwargs = call_kwargs
|
86
86
|
|
87
|
-
# Cached for performance.
|
87
|
+
# Cached for performance. Put kwargs in order of the call method instead
|
88
|
+
# of using the sorted key order from `tf.nest.flatten`.
|
88
89
|
self._flat_arguments = tf.nest.flatten(
|
89
|
-
(self.call_args, self.call_kwargs)
|
90
|
+
(self.call_args, self.call_kwargs.values())
|
90
91
|
)
|
91
92
|
# Used to avoid expensive `nest` operations in the most common case.
|
92
93
|
self._single_positional_tensor_passed = (
|
@@ -176,9 +177,13 @@ class Node:
|
|
176
177
|
for kt_id, kt_index in self._keras_inputs_ids_and_indices:
|
177
178
|
flat_arguments[kt_index] = tensor_dict[kt_id].pop()
|
178
179
|
|
180
|
+
# Pack the same way as `self._flat_arguments`, i.e. `kwargs` as a
|
181
|
+
# list in the original order.
|
179
182
|
args, kwargs = tf.nest.pack_sequence_as(
|
180
|
-
(self.call_args, self.call_kwargs), flat_arguments
|
183
|
+
(self.call_args, self.call_kwargs.values()), flat_arguments
|
181
184
|
)
|
185
|
+
# Add the keys to `kwargs` to go from a list to a dict.
|
186
|
+
kwargs = {k: v for k, v in zip(self.call_kwargs.keys(), kwargs)}
|
182
187
|
return args, kwargs
|
183
188
|
|
184
189
|
def serialize(self, make_node_key, node_conversion_map):
|
@@ -1,4 +1,4 @@
|
|
1
|
-
tf_keras/__init__.py,sha256=
|
1
|
+
tf_keras/__init__.py,sha256=iP91gG_iaiqS73EdUEnZwSHf6BqWoc6a3L38QmUHzXk,911
|
2
2
|
tf_keras/__internal__/__init__.py,sha256=OHQbeIC0QtRBI7dgXaJaVbH8F00x8dCI-DvEcIfyMsE,671
|
3
3
|
tf_keras/__internal__/backend/__init__.py,sha256=LnMs2A6685gDG79fxqmdulIYlVE_3WmXlBTBo9ZWYcw,162
|
4
4
|
tf_keras/__internal__/layers/__init__.py,sha256=F5SGMhOTPzm-PR44VrfinURHcVeQPIEdwnZlAkSTB3A,176
|
@@ -282,7 +282,7 @@ tf_keras/src/engine/functional_utils.py,sha256=5creFfo9UoG5OLJgkcw9gsfT-qch-RamT
|
|
282
282
|
tf_keras/src/engine/input_layer.py,sha256=QVAA9ZrhfUlcx0Tj_UuNF3t1nxYrhyks6vDJJeb18W8,18258
|
283
283
|
tf_keras/src/engine/input_spec.py,sha256=W3mojApaM_lN8Vr2MCvddE8RhHckxwObzpySPm16oEM,12076
|
284
284
|
tf_keras/src/engine/keras_tensor.py,sha256=rmIyf-sMKzGAMXzob0hCTZ3qA4JBYyIM85XUdmOPmqQ,28858
|
285
|
-
tf_keras/src/engine/node.py,sha256=
|
285
|
+
tf_keras/src/engine/node.py,sha256=mevKNFEtzeVbwLRuwB7sMzQGKt6ppIxLmMcfQMzu8N8,14254
|
286
286
|
tf_keras/src/engine/partial_batch_padding_handler.py,sha256=TNZvGXL-fvmZLLHIMPX_hy0w9LT8W52DHW7ZtnEvBvI,4325
|
287
287
|
tf_keras/src/engine/saving.py,sha256=So_T5PRjCOLzpOPGHNBiTCOrNvqvvGNK8AwZBwFQCbs,853
|
288
288
|
tf_keras/src/engine/sequential.py,sha256=UeW__ZHBL_lBgQfdyZZAOGW8KTjXy2dKsCu0IS3dR4Y,22974
|
@@ -606,7 +606,7 @@ tf_keras/src/utils/legacy/__init__.py,sha256=EfMmeHYDzwvxNaktPhQbkTdcPSIGCqMhBND
|
|
606
606
|
tf_keras/utils/__init__.py,sha256=b7_d-USe_EmLo02_P99Q1rUCzKBYayPCfiYFStP-0nw,2735
|
607
607
|
tf_keras/utils/experimental/__init__.py,sha256=DzGogE2AosjxOVILQBT8PDDcqbWTc0wWnZRobCdpcec,97
|
608
608
|
tf_keras/utils/legacy/__init__.py,sha256=7ujlDa5HeSRcth2NdqA0S1P2-VZF1kB3n68jye6Dj-8,189
|
609
|
-
tf_keras_nightly-2.19.0.
|
610
|
-
tf_keras_nightly-2.19.0.
|
611
|
-
tf_keras_nightly-2.19.0.
|
612
|
-
tf_keras_nightly-2.19.0.
|
609
|
+
tf_keras_nightly-2.19.0.dev2024101809.dist-info/METADATA,sha256=mQmZKjQp3wIKufIEvDuqmXi6wMAOFxw-Gip1-cnCgnY,1638
|
610
|
+
tf_keras_nightly-2.19.0.dev2024101809.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
|
611
|
+
tf_keras_nightly-2.19.0.dev2024101809.dist-info/top_level.txt,sha256=LC8FK7zHDNKxB17C6lGKvrZ_fZZGJsRiBK23SfiDegY,9
|
612
|
+
tf_keras_nightly-2.19.0.dev2024101809.dist-info/RECORD,,
|
File without changes
|