Searched refs:ConcatenateIota (Results 1 – 1 of 1) sorted by relevance
117 XlaOp ConcatenateIota(xla::XlaBuilder* b, XlaOp indices, in ConcatenateIota() function519 auto gather_indices = ConcatenateIota(ctx->builder(), top_left, warp_shape); in Compile()656 auto gather_indices = ConcatenateIota(ctx->builder(), top_left, warp_shape); in Compile()