Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "ad4125d1a9c4796cdbc6c6a5cdb69b09e60e5509"
Switch branch/tag
  1. 29 Dec, 2021 3 commits
    • Xin Wang's avatar
      Added block diagonal feedforward layer. · a7e60974
      Xin Wang authored 
      This layer replaces the weight matrix of the output_dense layer with a block diagonal matrix to save layer parameters and FLOPs. A linear mixing layer can be added optionally to improve layer expressibility.

PiperOrigin-RevId: 418828099
      a7e60974
    • Frederick Liu's avatar
      Internal change · 60f6d6cb
      Frederick Liu authored 
      PiperOrigin-RevId: 418817477
      60f6d6cb
    • Hongkun Yu's avatar
      Internal change · 5fea53a7
      Hongkun Yu authored 
      PiperOrigin-RevId: 418724903
      5fea53a7
  3. 28 Dec, 2021 2 commits
  5. 22 Dec, 2021 2 commits
  7. 21 Dec, 2021 3 commits
  9. 20 Dec, 2021 2 commits
  11. 17 Dec, 2021 2 commits
  13. 16 Dec, 2021 3 commits
  15. 15 Dec, 2021 5 commits
  17. 14 Dec, 2021 1 commit
  19. 13 Dec, 2021 7 commits
  21. 11 Dec, 2021 1 commit
  23. 10 Dec, 2021 1 commit
  25. 09 Dec, 2021 4 commits
  27. 08 Dec, 2021 1 commit
    • Smit Hinsu's avatar
      Replace dynamic Squeeze with reshape to 1d · ce7d8065
      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
      ce7d8065
  29. 07 Dec, 2021 3 commits