diff options
-rw-r--r-- | tensorflow/core/kernels/lookup_table_op.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/lookup_table_op.cc b/tensorflow/core/kernels/lookup_table_op.cc index f53b3ee3d2..7b46560425 100644 --- a/tensorflow/core/kernels/lookup_table_op.cc +++ b/tensorflow/core/kernels/lookup_table_op.cc @@ -863,6 +863,7 @@ REGISTER_KERNEL(string, int64); REGISTER_KERNEL(int64, string); REGISTER_KERNEL(int64, int64); REGISTER_KERNEL(string, string); +REGISTER_KERNEL(string, bool); #undef REGISTER_KERNEL @@ -879,6 +880,7 @@ REGISTER_KERNEL(string, string); REGISTER_KERNEL(string, float); REGISTER_KERNEL(string, int64); REGISTER_KERNEL(int64, string); +REGISTER_KERNEL(string, bool); #undef REGISTER_KERNEL @@ -895,6 +897,7 @@ REGISTER_KERNEL(int64, string); REGISTER_KERNEL(string, float); REGISTER_KERNEL(string, int64); REGISTER_KERNEL(int64, string); +REGISTER_KERNEL(string, bool); #undef REGISTER_KERNEL @@ -912,6 +915,7 @@ REGISTER_KERNEL(int64, int64); REGISTER_KERNEL(int64, float); REGISTER_KERNEL(int64, double); REGISTER_KERNEL(string, float); +REGISTER_KERNEL(string, bool); #undef REGISTER_KERNEL |