[XLA:GPU] Replace TF_RET_CHECKs with return Status.
[XLA:GPU] Replace TF_RET_CHECKs with return Status.

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/9319 from jaro-sevcik:no-cublas-or-triton-for-small-matmuls 690bb0e1d8b5894f64134d68a031d387399c638f
