Skip to content

GitLab

"IMG/cpio/ventoy/hook/openEuler/ventoy-timeout.sh" did not exist on "05a1b863a66bf72b26e5d87570c4e0e61b9736cd"
Switch branch/tag
Find file BlameHistoryPermalink
  • Tian Lin's avatar
    Merged commit that fixes transformer's predict and eval. (#6874) · b9cab01b
    Tian Lin authored 
    * Merged commit includes the following changes:
249776315  by tianlin<tianlin@google.com>:

    Internal change

249763206  by tianlin<tianlin@google.com>:

    For TF 2.0 (related to Beam Search), expand cond dims in tf.where(cond, x, y) to make all parameters broadcastable.

--
249392724  by hongkuny<hongkuny@google.com>:

    Internal change

PiperOrigin-RevId: 249776315

* Merged commit includes the following changes:
249823043  by tianlin<tianlin@google.com>:

    Bring back v2 test for predict and eval.

--

PiperOrigin-RevId: 249823043
    b9cab01b
mnist_tpu.py 7.29 KB