Visualize the graph of a custom model in Tensorflow 2.4

Question

I want to visualize my model graph in Tensorboard to check if I implemented my model correct.
I´m implementing my own model by subclassing the tf.keras.Layer