Searched refs:new_ddnums (Results 1 – 1 of 1) sorted by relevance
113 void UpdateDDNums(DotDimensionNumbers* new_ddnums, int64 reshaped_dim, in UpdateDDNums() argument129 update_dims(new_ddnums->mutable_lhs_contracting_dimensions()); in UpdateDDNums()130 update_dims(new_ddnums->mutable_lhs_batch_dimensions()); in UpdateDDNums()132 update_dims(new_ddnums->mutable_rhs_contracting_dimensions()); in UpdateDDNums()133 update_dims(new_ddnums->mutable_rhs_batch_dimensions()); in UpdateDDNums()754 DotDimensionNumbers new_ddnums; in PartitionBaseCase() local756 new_ddnums = original_hlo->dot_dimension_numbers(); in PartitionBaseCase()868 UpdateDDNums(&new_ddnums, slice_sharding_dim, true); in PartitionBaseCase()874 UpdateDDNums(&new_ddnums, slice_sharding_dim, false); in PartitionBaseCase()903 UpdateDDNums(&new_ddnums, lhs_concat_dim, true); in PartitionBaseCase()[all …]