Searched refs:diag_base_indices (Results 1 – 1 of 1) sorted by relevance
130 XlaOp diag_base_indices = BroadcastInDim(Iota(builder, S32, diag_len), in GetMatrixDiagonalViaGather() local135 XlaOp start_indices = Add(diag_base_indices, diag_offset); in GetMatrixDiagonalViaGather()