Skip to content

GitLab

Switch branch/tag
  1. 18 Jan, 2022 3 commits
  3. 12 Jan, 2022 1 commit
  5. 07 Jan, 2022 1 commit
  7. 04 Jan, 2022 8 commits
  9. 29 Dec, 2021 6 commits
  11. 28 Dec, 2021 4 commits
  13. 21 Dec, 2021 2 commits
  15. 16 Dec, 2021 2 commits
  17. 15 Dec, 2021 2 commits
  19. 13 Dec, 2021 4 commits
  21. 08 Dec, 2021 2 commits
    • 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
    • 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
  23. 07 Dec, 2021 4 commits
  25. 03 Dec, 2021 1 commit