add symbolic args changes as well

Author: divyashreepathihalli

keras/src/ops/symbolic_arguments.py

@@ -36,11 +36,13 @@ def fill_in(self, tensor_dict):
         if self._single_positional_tensor is not None:
             # Performance optimization for most common case.
             # Approx. 70x faster.
-            return (tensor_dict[id(self._single_positional_tensor)],), {}
+            return (tensor_dict.get(id(self._single_positional_tensor), None),), {}
 
         def switch_fn(x):
             if isinstance(x, KerasTensor):
-                return tensor_dict.get(id(x), None)
+                result = tensor_dict.get(id(x), None)
+                print(f"[DEBUG] SymbolicArguments.fill_in: tensor_id={id(x)}, result={result}")
+                return result
             return x
 
         return self.convert(switch_fn)