official/nlp/modeling/layers/transformer.py · 7ebfc3dddeb85fd752c083b657412c214fdc3fd3 · ModelZoo / ResNet50_tensorflow

"vscode:/vscode.git/clone" did not exist on "8b4cb777db4bd72a25de9ffc77ba7f9facb9ee9a"

Enable XLA compilation using `@tf.function(experimental_compile=True) for transformer layer. · 393c1399

George Karpenkov authored Feb 21, 2020

To debug the tf.function this API can be used: https://www.tensorflow.org/api_docs/python/tf/config/experimental_run_functions_eagerly

PiperOrigin-RevId: 296458870

393c1399

transformer.py 10 KB

Replace transformer.py