message: Add bfloat16 support to sparse_tensor_dense_matmul_op.
  This op is an effective replacement in some circumstances for embedding_lookup_sparse, which can handle bfloat16 embeddings.
diff: @@ -23,6 +23,7 @@ limitations under the License.
  #include "tensorflow/core/framework/op.h"
  #include "tensorflow/core/framework/op_kernel.h"
  #include "tensorflow/core/kernels/fill_functor.h"
  +#include "tensorflow/core/lib/bfloat16/bfloat16.h"
  namespace tensorflow {
  @@ -173,6 +174,7 @@ REGISTER_KERNELS_CPU(double);
  REGISTER_KERNELS_CPU(int32);
  REGISTER_KERNELS_CPU(complex64);
  REGISTER_KERNELS_CPU(complex128);
  +REGISTER_KERNELS_CPU(bfloat16);
  #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  