Skip to content

GitLab

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
optimizer.py 6.03 KB