From f508f77fb2f4fd596ac0bcfbd7bb57ee432da143 Mon Sep 17 00:00:00 2001
From: Dero Gharibian <dero@google.com>
Date: Mon, 18 Nov 2019 15:32:54 -0800
Subject: [PATCH] Migrate from std::string to tensorflow::tstring.

Note that during the transition period tstring is typedef'ed to
std::string.

See: https://github.com/tensorflow/community/pull/91
---
 .../custom_ops/text/cc/kernels/skip_gram_kernels.cc           | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tensorflow_addons/custom_ops/text/cc/kernels/skip_gram_kernels.cc b/tensorflow_addons/custom_ops/text/cc/kernels/skip_gram_kernels.cc
index 4f1caa6da2..5b38d0665d 100644
--- a/tensorflow_addons/custom_ops/text/cc/kernels/skip_gram_kernels.cc
+++ b/tensorflow_addons/custom_ops/text/cc/kernels/skip_gram_kernels.cc
@@ -133,7 +133,7 @@ class SkipGramGenerateCandidatesOp : public OpKernel {
                               .TypeConstraint<type>("T"),           \
                           SkipGramGenerateCandidatesOp<type>)
 
-REGISTER_KERNEL(string);
+REGISTER_KERNEL(tstring);
 REGISTER_KERNEL(int64);
 REGISTER_KERNEL(int32);
 REGISTER_KERNEL(int16);
@@ -141,4 +141,4 @@ REGISTER_KERNEL(int16);
 #undef REGISTER_KERNEL
 
 }  // end namespace addons
-}  // namespace tensorflow
\ No newline at end of file
+}  // namespace tensorflow