Oops, sorry this is not an answer and I don't know how to delete this.
Thank you for posting this. I have encountered the same problem. It also happens on tf.argsort()
data = tf.random.uniform([20])
argsorted = tf.argsort(data, 0, stable=True)
print(f'{argsorted=}')
output:
argsorted=<tf.Tensor: shape=(20,), dtype=int32, numpy=array([ 4, 17, 9, 1, 7, 12, 8, 14, 3, 0, 18, 13, 5, 10, 15, 19, 0, 0, 0, 0], dtype=int32)>
I installed tensorflow, following the exact path you have pointed out.