"tests/unit/ops/sparse_attention/test_sparse_attention.py" did not exist on "1b2721adcd96656bb1f27d1f2f60947567b2d505"
  1. 18 Jan, 2022 1 commit
  2. 12 Jan, 2022 1 commit
  3. 07 Jan, 2022 1 commit
  4. 04 Jan, 2022 4 commits
  5. 29 Dec, 2021 3 commits
  6. 28 Dec, 2021 2 commits
  7. 21 Dec, 2021 1 commit
  8. 16 Dec, 2021 1 commit
  9. 15 Dec, 2021 1 commit
  10. 13 Dec, 2021 2 commits
  11. 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
  12. 07 Dec, 2021 2 commits
  13. 03 Dec, 2021 2 commits
  14. 01 Dec, 2021 2 commits
  15. 30 Nov, 2021 1 commit
  16. 29 Nov, 2021 2 commits
  17. 25 Nov, 2021 1 commit
  18. 23 Nov, 2021 2 commits
  19. 22 Nov, 2021 2 commits
  20. 19 Nov, 2021 1 commit
  21. 17 Nov, 2021 1 commit
  22. 16 Nov, 2021 1 commit
  23. 15 Nov, 2021 2 commits
  24. 10 Nov, 2021 3 commits