Skip to content

GitLab

Switch branch/tag
  1. 08 Dec, 2021 1 commit
    • Smit Hinsu's avatar
      Replace dynamic Squeeze with reshape to 1d · ea36aff1
      Smit Hinsu authored 
      TPUs don't support dynamic Squeeze as it won't be able to identify the dimensions to remove if the input shape is dynamic. For example, [<=16, 1] could return a vector or scalar result depending on the size of the dynamic dim.

PiperOrigin-RevId: 415083667
      ea36aff1
  3. 17 Nov, 2021 1 commit
  5. 12 Apr, 2021 2 commits
    • Reed Wanderman-Milne's avatar
      Use nonexperimental mixed precision API for official models. · ba8ad4f5
      Reed Wanderman-Milne authored 
      For all modified calls to set_mixed_precision_policy(), the loss_scale argument was removed, as it cannot be passed if the nonexperimental API is used. For all such callers, the loss_scale is later used to explicitly create a LossScaleOptimizer, so removing the argument has no impact.

Switching to the non-experimental LossScaleOptimizer has no effect, as it has near identical behavior and all isinstance checks within the official models check for the non-experimental version.

PiperOrigin-RevId: 368101975
      ba8ad4f5
    • Reed Wanderman-Milne's avatar
      Use nonexperimental mixed precision API for official models. · 0d8f9807
      Reed Wanderman-Milne authored 
      For all modified calls to set_mixed_precision_policy(), the loss_scale argument was removed, as it cannot be passed if the nonexperimental API is used. For all such callers, the loss_scale is later used to explicitly create a LossScaleOptimizer, so removing the argument has no impact.

Switching to the non-experimental LossScaleOptimizer has no effect, as it has near identical behavior and all isinstance checks within the official models check for the non-experimental version.

PiperOrigin-RevId: 368101975
      0d8f9807
  7. 10 Mar, 2021 2 commits
  9. 15 Dec, 2020 2 commits
  11. 16 Nov, 2020 2 commits
  13. 14 Nov, 2020 2 commits
  15. 26 Oct, 2020 2 commits
  17. 30 Sep, 2020 2 commits
  19. 13 Sep, 2020 2 commits
  21. 21 Aug, 2020 2 commits
  23. 14 Aug, 2020 1 commit
  25. 12 Aug, 2020 2 commits
  27. 07 Aug, 2020 1 commit
  29. 22 Jul, 2020 2 commits
  31. 12 Jun, 2020 2 commits
  33. 04 Jun, 2020 1 commit
  35. 03 Jun, 2020 1 commit
  37. 28 May, 2020 1 commit
  39. 26 May, 2020 1 commit
  41. 23 May, 2020 1 commit
  43. 19 May, 2020 1 commit
  45. 18 May, 2020 1 commit
  47. 04 May, 2020 1 commit
  49. 22 Apr, 2020 1 commit
  51. 20 Apr, 2020 1 commit
  53. 17 Apr, 2020 2 commits