This repository has been archived by the owner on Jan 21, 2025. It is now read-only.

Option to use mtf.Print to log which tokens are sent to which experts when run on CPU.#329

Open

copybara-service[bot] wants to merge 1 commit intomasterfrom test_368137313

+85-11

Commits on Nov 9, 2021

Option to use mtf.Print to log which tokens are sent to which experts when run on CPU.

William Fedus
authored and
Mesh TensorFlow Team
committed