onnxruntime
62383b03 - Improve BERT optimization script: Gelu and LayerNorm for models from Tensorflow 2.* (#2957)

Commit
5 years ago
Improve BERT optimization script: Gelu and LayerNorm for models from Tensorflow 2.* (#2957) * Add unit test. Add an option --use_onnxruntime to use onnxruntime to do optimization for pytorch model. Update layer norm and gelu for tensorflow 2.1 keras bert model. Add logging and use f-strings. Add extra checking for tensorflow model reshape fusion. Allow output model to json for test purpose. update match parent path utility function to return index * remove function not used.
Author
Parents
Loading