Remove the mocking of LayerNormalization in tests

[ondrasej]

As discussed in the review of #132, we're patching tf_keras.layers.LayerNormalization with itself in some of our tests, e.g. in gematria/granite/python/gnn_model_base_test.py. We should investigate whether this mocking is still needed, and if not, remove it.