We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 055113f commit 987f455Copy full SHA for 987f455
tensorflow_addons/custom_ops/text/cc/kernels/skip_gram_kernels.cc
@@ -133,12 +133,12 @@ class SkipGramGenerateCandidatesOp : public OpKernel {
133
.TypeConstraint<type>("T"), \
134
SkipGramGenerateCandidatesOp<type>)
135
136
-REGISTER_KERNEL(string);
+REGISTER_KERNEL(tstring);
137
REGISTER_KERNEL(int64);
138
REGISTER_KERNEL(int32);
139
REGISTER_KERNEL(int16);
140
141
#undef REGISTER_KERNEL
142
143
} // end namespace addons
144
-} // namespace tensorflow
+} // namespace tensorflow
0 commit comments