Skip to content

Commit 987f455

Browse files
gharibianseanpmorgan
authored andcommitted
Migrate from std::string to tensorflow::tstring. (#705)
Note that during the transition period tstring is typedef'ed to std::string. See: tensorflow/community#91
1 parent 055113f commit 987f455

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorflow_addons/custom_ops/text/cc/kernels/skip_gram_kernels.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -133,12 +133,12 @@ class SkipGramGenerateCandidatesOp : public OpKernel {
133133
.TypeConstraint<type>("T"), \
134134
SkipGramGenerateCandidatesOp<type>)
135135

136-
REGISTER_KERNEL(string);
136+
REGISTER_KERNEL(tstring);
137137
REGISTER_KERNEL(int64);
138138
REGISTER_KERNEL(int32);
139139
REGISTER_KERNEL(int16);
140140

141141
#undef REGISTER_KERNEL
142142

143143
} // end namespace addons
144-
} // namespace tensorflow
144+
} // namespace tensorflow

0 commit comments

Comments
 (0)