Searched refs:iota_s32 (Results 1 – 1 of 1) sorted by relevance
32 XlaOp iota_s32 = Iota(builder, iota_shape, last_dim); in TopK() local37 Sort({Neg(input), iota_s32}, in TopK()39 iota_s32.builder()), in TopK()