message: Do not use SparseMatmul to for bfloat16 as Matmul is already supported.
diff: @@ -2093,8 +2093,9 @@ def matmul(a,
  sparse_matmul_types = [dtypes.bfloat16, dtypes.float32]
  use_sparse_matmul = (
  a.dtype in sparse_matmul_types and b.dtype in sparse_matmul_types)
  - if a.dtype == dtypes.bfloat16 or b.dtype == dtypes.bfloat16:
  - # matmul currently doesn't handle bfloat16 inputs.
  + if (a.dtype == dtypes.bfloat16 or b.dtype == dtypes.bfloat16 and
  + a.dtype != b.dtype):
  + # matmul currently doesn't handle mixed-precision inputs.
  use_sparse_matmul = True
  if use_sparse_matmul:
  ret = sparse_matmul(
  