Searched refs:diag_offset (Results 1 – 1 of 1) sorted by relevance
132 XlaOp diag_offset = in GetMatrixDiagonalViaGather() local135 XlaOp start_indices = Add(diag_base_indices, diag_offset); in GetMatrixDiagonalViaGather()