Searched defs:iota_s32 (Results 1 – 1 of 1) sorted by relevance
50 XlaOp iota_s32 = Iota(builder, iota_shape, last_dim); in TopK() local92 XlaOp iota_s32 = Iota(builder, iota_shape, last_dim); in TopKWithPartitions() local106 auto iota_s32 = values_and_indices[3]; in TopKWithPartitions() local