Searched refs:dot_shape_r2 (Results 1 – 1 of 1) sorted by relevance
91 Shape dot_shape_r2 = ShapeUtil::MakeShape(dot_type, {dot_rows, dot_cols}); in DecomposeBatchDot() local140 HloInstruction::CreateDot(dot_shape_r2, lhs_slice_r2, rhs_slice_r2, in DecomposeBatchDot()