The inputs are used for V1 layer's losses (`layer.get_losses_for(input)`), but this code is also causing issues when tracing layers with multiple inputs. This change fixes the tracing errors. PiperOrigin-RevId: 358235121 Change-Id: I94b667d1f89be2415e046ebeb00c4adc1acdc207 |
||
---|---|---|
.. | ||
saved_model | ||
__init__.py | ||
BUILD | ||
hdf5_format.py | ||
losses_serialization_test.py | ||
metrics_serialization_test.py | ||
model_config.py | ||
save_test.py | ||
save_weights_test.py | ||
save.py | ||
saved_model_experimental_test.py | ||
saved_model_experimental.py | ||
saving_utils_test.py | ||
saving_utils.py |