Searched refs:new_sort_shape (Results 1 – 2 of 2) sorted by relevance
78 Shape new_sort_shape = new_shapes.size() == 1 in RemoveUnusedOperandFromSort() local82 sort->CloneWithNewOperands(new_sort_shape, operands)); in RemoveUnusedOperandFromSort()
105 Shape new_sort_shape = ShapeUtil::MakeTupleShape(new_shapes); in ExpandInstruction() local107 sort->CloneWithNewOperands(new_sort_shape, new_operands)); in ExpandInstruction()