Add support for dynamic dimensions for ops handling tf.IndexedSlices. (#21148)

hertschuh · web-flow · commit 78ca6a46b690 · 2025-04-09T14:35:33.000-07:00
Fixes #21069
diff --git a/keras/src/backend/tensorflow/sparse.py b/keras/src/backend/tensorflow/sparse.py
@@ -131,7 +131,7 @@ def values_for_union(indices_expanded, indices_count, values):
         )
         to_union_indices = tf.gather(indices_indices, union_indices)
         values_with_leading_zeros = tf.concat(
-            [tf.zeros((1,) + values.shape[1:], values.dtype), values], axis=0
+            [tf.zeros_like(values[0:1]), values], axis=0
         )
         return tf.gather(values_with_leading_zeros, to_union_indices)
 

Original file line number	Diff line number	Diff line change
`@@ -131,7 +131,7 @@ def values_for_union(indices_expanded, indices_count, values):`
`131`	`131`	`)`
`132`	`132`	`to_union_indices = tf.gather(indices_indices, union_indices)`
`133`	`133`	`values_with_leading_zeros = tf.concat(`
`134`		`- [tf.zeros((1,) + values.shape[1:], values.dtype), values], axis=0`
	`134`	`+ [tf.zeros_like(values[0:1]), values], axis=0`
`135`	`135`	`)`
`136`	`136`	`return tf.gather(values_with_leading_zeros, to_union_indices)`
`137`	`137`