Skip to content

Add unit test for EmbedLayerNormalization quantization op.#8033

Merged
nkreeger merged 9 commits intomasterfrom
kreeger/embed_layernorm_test.py
Jun 11, 2021
Merged

Add unit test for EmbedLayerNormalization quantization op.#8033
nkreeger merged 9 commits intomasterfrom
kreeger/embed_layernorm_test.py

Conversation

@nkreeger
Copy link
Contributor

While working on EmbedLayerNormalization optimization I noticed that the quantization script did not have a unit test. This change simple introduces a unit test which validates changes to the graph and ensures that the model still executes.

@nkreeger nkreeger requested a review from yufenglee June 11, 2021 11:58
@nkreeger nkreeger requested a review from a team as a code owner June 11, 2021 11:58
@nkreeger nkreeger merged commit 1d7f44a into master Jun 11, 2021
@nkreeger nkreeger deleted the kreeger/embed_layernorm_test.py branch June 11, 2021 22:33
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants