Searched refs:source_indices (Results 1 – 2 of 2) sorted by relevance
825 std::vector<int64> source_indices(indices.begin(), indices.end()); in ExchangeHalo() local826 source_indices[dim] -= i + 1; in ExchangeHalo()828 target.tile_assignment()(source_indices), device); in ExchangeHalo()
512 DimensionVector source_indices(rank, 0); in HandleConcatenate() local518 GetEvaluatedLiteralFor(operand), source_indices, dest_indices, in HandleConcatenate()