Searched refs:slice_sharding_dim (Results 1 – 1 of 1) sorted by relevance
719 int64 slice_sharding_dim = -1; in PartitionBaseCase() local723 slice_sharding_dim = i; in PartitionBaseCase()731 rhs_concat_dim = slice_sharding_dim; in PartitionBaseCase()733 lhs_concat_dim = slice_sharding_dim; in PartitionBaseCase()738 ? indices_map.rhs_to_lhs_indices[slice_sharding_dim] in PartitionBaseCase()739 : slice_sharding_dim; in PartitionBaseCase()742 ? slice_sharding_dim in PartitionBaseCase()743 : indices_map.lhs_to_rhs_indices[slice_sharding_dim]; in PartitionBaseCase()747 indices_map.output_to_lhs_indices[slice_sharding_dim]; in PartitionBaseCase()750 indices_map.output_to_rhs_indices[slice_sharding_dim]; in PartitionBaseCase()[all …]