Searched refs:sort_op (Results 1 – 4 of 4) sorted by relevance
55 } else if (auto sort_op = llvm::dyn_cast<SortOp>(op)) { in runOnFunction() local56 SinkToRegion(&sort_op.comparator()); in runOnFunction()
494 auto sort_op = func_builder->create<mlir::mhlo::SortOp>( in ImportInstructionImpl() local499 ImportAsRegion(*sort_instruction->to_apply(), &sort_op.comparator())); in ImportInstructionImpl()503 return sort_op.getOperation(); in ImportInstructionImpl()507 ->create<mlir::mhlo::TupleOp>(loc, result_type, sort_op.getResults()) in ImportInstructionImpl()
2856 auto sort_op = mlir::cast<mlir::lmhlo::SortOp>(mlir_input.op); in EmitSortFromMlir() local2858 context.SetOperation(sort_op); in EmitSortFromMlir()2863 int64 dimension_to_sort = sort_op.dimension(); in EmitSortFromMlir()2875 GetAllocationSliceForMlir(sort_op.output()[i])); in EmitSortFromMlir()2877 GetAllocationSliceForMlir(sort_op.operands()[i])); in EmitSortFromMlir()2992 BuildKernelThunkForMlir(sort_op, sort_op.output(), Thunk::ThunkInfo(), in EmitSortFromMlir()3006 &sort_op.comparator(), /*is_fusion=*/false)); in EmitSortFromMlir()
4757 auto sort_op = rewriter.create<mhlo::SortOp>( in matchAndRewrite() local4768 &sort_op.comparator(), &rewriter); in matchAndRewrite()4771 auto tuple_first_element = sort_op.getResult(0); in matchAndRewrite()4772 auto tuple_second_element = sort_op.getResult(1); in matchAndRewrite()