I am trying to load weights and for that to work i need to perform the following:
dummy_input = tf.random.uniform(input