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