Searched refs:dot_shape_r3 (Results 1 – 1 of 1) sorted by relevance
92 Shape dot_shape_r3 = ShapeUtil::MakeShape(dot_type, {1, dot_rows, dot_cols}); in DecomposeBatchDot() local145 HloInstruction::CreateReshape(dot_shape_r3, dot_r2)); in DecomposeBatchDot()