diff options
author | 2017-02-05 11:59:39 -0800 | |
---|---|---|
committer | 2017-02-05 12:09:42 -0800 | |
commit | 824629120293f92bcaba07fd33d7e6546b5804a5 (patch) | |
tree | 85f492f41358bcf425741bbe271ce0f2109ace45 | |
parent | 653e35ae589740fbb9c572e0548427be985a7515 (diff) |
Allow lookup_table_ops to have bools as values.
Change: 146608995
-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 |