PR #24004: Remove unused cub headers.

Imported from GitHub PR #24004

Copybara import of the project:

--
0fd44ae by Dimitris Vardoulakis <[email protected]>:

Remove unused cub headers.

Merging this change closes #24004

COPYBARA_INTEGRATE_REVIEW=#24004 from dimvar:remove-unused-cubsort-headers 0fd44ae
PiperOrigin-RevId: 739109352

Author: dimvar

xla/service/gpu/gpu_prim.h

@@ -27,8 +27,6 @@ limitations under the license, the license you must see.
 #include "cub/device/device_segmented_radix_sort.cuh"
 #include "cub/device/device_segmented_reduce.cuh"
 #include "cub/device/device_select.cuh"
-#include "cub/iterator/counting_input_iterator.cuh"
-#include "cub/iterator/transform_input_iterator.cuh"
 #include "cub/thread/thread_operators.cuh"
 #include "cub/warp/warp_reduce.cuh"
 #include "third_party/gpus/cuda/include/cusparse.h"