Home
last modified time | relevance | path

Searched refs:new_sort (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dsort_simplifier.cc81 HloInstruction* new_sort = computation->AddInstruction( in RemoveUnusedOperandFromSort() local
108 new_sort->set_to_apply(new_compare); in RemoveUnusedOperandFromSort()
112 if (new_sort->shape().IsTuple()) { in RemoveUnusedOperandFromSort()
119 *new_shapes[new_index], new_sort, new_index)); in RemoveUnusedOperandFromSort()
125 result_map[*used_indices.begin()] = new_sort; in RemoveUnusedOperandFromSort()
H A Dstable_sort_expander.cc106 HloInstruction* new_sort = computation->AddInstruction( in ExpandInstruction() local
118 sort->operand(i)->shape(), new_sort, i))); in ExpandInstruction()
125 sort = Cast<HloSortInstruction>(new_sort); in ExpandInstruction()