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 /Eigen/src/Core/arch/NEON/PacketMath.h | |
| 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 'Eigen/src/Core/arch/NEON/PacketMath.h')
0 files changed, 0 insertions, 0 deletions