xla

a008de35 - Update op argmin and argmax from manual lowering to codegen. (#5586)