Searched refs:rhs_r3 (Results 1 – 1 of 1) sorted by relevance
99 auto rhs_r3 = computation->AddInstruction( in DecomposeBatchDot() local114 HloInstruction::CreateSlice(rhs_slice_shape_r3, rhs_r3, {i, 0, 0}, in DecomposeBatchDot()