keras.utils.plot_model continua me dizendo para instalar pydot e graphviz
tf.keras.utils.plot_model(model, to_file='model_test.png', show_shapes=True, show_layer_names=True)
Trisiegt