Searched refs:output_to_rhs_indices (Results 1 – 1 of 1) sorted by relevance
110 std::vector<int64> output_to_rhs_indices; member347 std::vector<int64> output_to_rhs_indices(output_rank, -1); in ComputeDimensionIndexMapping() local360 output_to_rhs_indices[mapping.output] = mapping.rhs; in ComputeDimensionIndexMapping()380 output_to_lhs_indices, output_to_rhs_indices}; in ComputeDimensionIndexMapping()509 indices_map.output_to_rhs_indices); in PartitionBaseCase()516 output_sharding, indices_map.output_to_rhs_indices, in PartitionBaseCase()750 indices_map.output_to_rhs_indices[slice_sharding_dim]; in PartitionBaseCase()2327 : indices_map.output_to_rhs_indices; in LhsIsBestMatchForNonContractingPartitioning()2329 assume_lhs_match ? indices_map.output_to_rhs_indices in LhsIsBestMatchForNonContractingPartitioning()