print input_layer._outbound_nodes

Dobiasd · Apr 10, 2024 · e917410 · e917410
1 parent 925ea2c
commit e917410
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/keras_export/convert_model.py b/keras_export/convert_model.py
@@ -167,7 +167,8 @@ def set_shape_idx_0_to_1_if_none(shape):
 
     def generate_input_data(input_layer):
         """Random data fitting the input shape of a layer."""
-        print("input layer type", type(input_layer).__name__)  # todo: remove
+        print("input input_layer type", type(input_layer).__name__)  # todo: remove
+        print("input_layer._outbound_nodes type", type(input_layer._outbound_nodes).__name__)  # todo: remove
         if input_layer._outbound_nodes and isinstance(
                 get_first_outbound_op(input_layer), Embedding):
             random_fn = lambda size: np.random.randint(