Searched refs:dot_r2 (Results 1 – 1 of 1) sorted by relevance
139 auto dot_r2 = computation->AddInstruction( in DecomposeBatchDot() local145 HloInstruction::CreateReshape(dot_shape_r3, dot_r2)); in DecomposeBatchDot()