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