Broken ExtensionType interoperability with Keras in 2.16 #67334
Labels
comp:keras
Keras related issues
stale
This label marks the issue/pr stale - to be closed automatically if no activity
stat:awaiting response
Status - Awaiting response from author
TF 2.16
type:bug
Bug
Issue type
Bug
Have you reproduced the bug with TensorFlow Nightly?
No
Source
binary
TensorFlow version
v2.16.1-0-g5bc9d26649c 2.16.1
Custom code
No
OS platform and distribution
MacOS 14.4.1 (Arm)
Mobile device
No response
Python version
No response
Bazel version
No response
GCC/compiler version
No response
CUDA/cuDNN version
No response
GPU model and memory
No response
Current behavior?
The ExtensionType guide suggests that it is possible to create
tf.keras.layers.Input
for ExtensionTypes using thetype_spec
argument. This argument, however, seems to be missing in Keras 3 / TF 2.16. It is not clear how the integration with Keras should work in 2.16 (if it is even possible due to the absence of the critical argument).Standalone code to reproduce the issue
Relevant log output
The text was updated successfully, but these errors were encountered: