Searched refs:sort_instr (Results 1 – 2 of 2) sorted by relevance
151 for (HloInstruction* sort_instr : sort_instrs) { in Run()152 TF_ASSIGN_OR_RETURN(bool result, RemoveUnusedOperandFromSort(sort_instr)); in Run()
419 auto* sort_instr = xla::Cast<xla::HloSortInstruction>(instr); in EmitSortOp() local420 sort.dimensionAttr(builder_.getI64IntegerAttr(sort_instr->sort_dimension())); in EmitSortOp()421 sort.is_stableAttr(builder_.getBoolAttr(sort_instr->is_stable())); in EmitSortOp()423 *sort_instr->called_computations()[0], &sort.comparator(), &builder_)); in EmitSortOp()