Fix searchsorted for multi-dimensional values in TF backend (#23436)