Always lower ragged dot for cpu, gpu, and tpu

PiperOrigin-RevId: 780546970

Author: vwbaker

jax/_src/lax/lax.py

@@ -6372,16 +6372,11 @@ def use_default_lowering():
   return [result]
 
 
-mlir.register_lowering(ragged_dot_general_p,
-                       mlir.lower_fun(_ragged_dot_general_impl,
-                                      multiple_results=False))
-
-for platform in ['tpu']:
-  mlir.register_lowering(
-      ragged_dot_general_p,
-      partial(_ragged_dot_general_lower, platform=platform),
-      platform=platform,
-  )
+mlir.register_lowering(
+    ragged_dot_general_p,
+    partial(_ragged_dot_general_lower, platform=platform),
+    platform=platform,
+)
 
 
 def _broadcast_in_dim_shape_rule(operand, *, shape, broadcast_dimensions,