Searched refs:rhs_slice_r3 (Results 1 – 1 of 1) sorted by relevance
113 auto rhs_slice_r3 = computation->AddInstruction( in DecomposeBatchDot() local117 HloInstruction::CreateReshape(rhs_slice_shape_r2, rhs_slice_r3)); in DecomposeBatchDot()