Google Git
Sign in
eigen / mirror / 3a087ccb99b454dc34484333e608e836e7032213
commit3a087ccb99b454dc34484333e608e836e7032213[log] [tgz]
authorAntonio Sanchez <cantonios@google.com>Fri Jun 25 14:22:19 2021 -0700
committerAntonio Sanchez <cantonios@google.com>Tue Jun 29 10:36:20 2021 -0700
treec14d2a34f8582ead6d7e9b24ceb3ba980bba3d4f
parent2d132d17365ffc84c0cc7a7da9b8f7090e94b476 [diff]
Modify tensor argmin/argmax to always return first occurence.

As written, depending on multithreading/gpu, the returned index from
`argmin`/`argmax` is not currently stable.  Here we modify the functors
to always keep the first occurence (i.e. if the value is equal to the
current min/max, then keep the one with the smallest index).

This is otherwise causing unpredictable results in some TF tests.
1 file changed
tree: c14d2a34f8582ead6d7e9b24ceb3ba980bba3d4f
  1. .gitlab/
  2. bench/
  3. blas/
  4. ci/
  5. cmake/
  6. debug/
  7. demos/
  8. doc/
  9. Eigen/
  10. failtest/
  11. lapack/
  12. scripts/
  13. test/
  14. unsupported/
  15. .gitignore
  16. .gitlab-ci.yml
  17. .hgeol
  18. CMakeLists.txt
  19. COPYING.APACHE
  20. COPYING.BSD
  21. COPYING.GPL
  22. COPYING.LGPL
  23. COPYING.MINPACK
  24. COPYING.MPL2
  25. COPYING.README
  26. CTestConfig.cmake
  27. CTestCustom.cmake.in
  28. eigen3.pc.in
  29. INSTALL
  30. README.md
  31. signature_of_eigen3_matrix_library
README.md

Eigen is a C++ template library for linear algebra: matrices, vectors, numerical solvers, and related algorithms.

For more information go to http://eigen.tuxfamily.org/.

For pull request, bug reports, and feature requests, go to https://gitlab.com/libeigen/eigen.