Simplify the structure of embedding, layer_normalization, and multi_head_attention tests.#642

copybara-service[bot] merged 1 commit intomastertensorflow/privacy:masterfrom

cl/908775823tensorflow/privacy:cl/908775823Copy head branch name to clipboard