official/nlp/modeling/layers/transformer.py · 393c1399dc73fed4eb359b7dd1b4e7d04e3216a5 · ModelZoo / ResNet50_tensorflow

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