Fix: Resolve infinite recursion in keras.ops.associative_scan under Tensorflow static graph mode (#22072)
* Fixed issue #22058 * solved issue #22058 * issue #22058 fixed * fixed issue #22058 * solved issue #22058 * modified code to handle <2 elem_length cases * format corrected * 'mutable' is explicitly set in nnx_metadata. * reformatted * formatted * formatted * formatted * solved the error for imutable vars * iterative approach * reformats * solved problems from review, removed dead code and used in-place mutation approach * required changes and critical bugs solved. Writing tests remain * reverted _direct_assign and _initialize, added tests * Added tests for dynamic cases using tf.TensorSpec(shape=[None, ...]) * reformats * used KerasTensor instead of tf.TensorSpec * solved the 5 issues * remaining fixes * fixes in _interleave_with_b * format fixes * fixed crash issue * format fixes
C
Chirag Wattamwar committed
76e271c9b5e53714878c492229dfb996f57b15f8
Parent: 0aaaa3d
Committed by GitHub <[email protected]>
on 3/30/2026, 8:24:03 PM