Update deprecated functions

[DimitrisCC]

Do you have a strategy on updating the tf.layers.dense calls to tf.keras.layers.Dense?
The latter one doesn't have a reuse parameter, so I guess the code should be refactored as a whole?

[yagmuronay]

Hi @DimitrisCC, do you have by any chance the refactored code? Thank you in advance!