diff options
author | Antonio Sanchez <cantonios@google.com> | 2021-06-25 14:22:19 -0700 |
---|---|---|
committer | Antonio Sanchez <cantonios@google.com> | 2021-06-29 10:36:20 -0700 |
commit | 3a087ccb99b454dc34484333e608e836e7032213 (patch) | |
tree | c14d2a34f8582ead6d7e9b24ceb3ba980bba3d4f /doc/snippets/HessenbergDecomposition_compute.cpp | |
parent | 2d132d17365ffc84c0cc7a7da9b8f7090e94b476 (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.
Diffstat (limited to 'doc/snippets/HessenbergDecomposition_compute.cpp')
0 files changed, 0 insertions, 0 deletions