Skip to content

GitLab

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