Searched refs:rhs_to_output_indices (Results 1 – 1 of 1) sorted by relevance
108 std::vector<int64> rhs_to_output_indices; member226 const std::vector<int64>& rhs_to_output_indices, in GenNewConvDNums() argument289 : rhs_to_output_indices[rhs_concat_dim]; in GenNewConvDNums()345 std::vector<int64> rhs_to_output_indices(rhs_rank, -1); in ComputeDimensionIndexMapping() local356 rhs_to_output_indices[mapping.rhs] = mapping.output; in ComputeDimensionIndexMapping()379 rhs_to_lhs_indices, rhs_to_output_indices, in ComputeDimensionIndexMapping()508 rhs_sharding, indices_map.rhs_to_output_indices, in PartitionBaseCase()517 indices_map.rhs_to_output_indices); in PartitionBaseCase()945 : indices_map.rhs_to_output_indices[rhs_concat_dim]; in PartitionBaseCase()990 indices_map.rhs_to_output_indices, new_dot_shape), in PartitionBaseCase()[all …]