diff options
author | Yu-Cheng Ling <ycling@google.com> | 2017-11-10 22:14:36 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-10 22:19:15 -0800 |
commit | 07ac134a4b2fa6f40f7fa8f7266b854a529ddcab (patch) | |
tree | 2346c92b54829b490d862244458b730d92decc33 | |
parent | 01e8710e3902a26692ac8bec32e41a8ffbf3b110 (diff) |
Fix //tensorflow/contrib/lite/tools:gen_op_registration bazel build.
PiperOrigin-RevId: 175376892
-rw-r--r-- | tensorflow/contrib/lite/tools/BUILD | 1 | ||||
-rw-r--r-- | tensorflow/contrib/lite/tools/gen_op_registration.cc | 3 | ||||
-rw-r--r-- | tensorflow/contrib/lite/tools/gen_op_registration.h | 1 |
3 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/tools/BUILD b/tensorflow/contrib/lite/tools/BUILD index 2d918fd4e8..3c9cec5d16 100644 --- a/tensorflow/contrib/lite/tools/BUILD +++ b/tensorflow/contrib/lite/tools/BUILD @@ -20,6 +20,7 @@ cc_library( hdrs = ["gen_op_registration.h"], deps = [ "//tensorflow/contrib/lite:framework", + "//tensorflow/contrib/lite:string", "@com_googlesource_code_re2//:re2", ], ) diff --git a/tensorflow/contrib/lite/tools/gen_op_registration.cc b/tensorflow/contrib/lite/tools/gen_op_registration.cc index 57c2567e3b..d80ea59170 100644 --- a/tensorflow/contrib/lite/tools/gen_op_registration.cc +++ b/tensorflow/contrib/lite/tools/gen_op_registration.cc @@ -15,8 +15,9 @@ limitations under the License. #include <string> #include <vector> -#include "third_party/re2/re2.h" +#include "re2/re2.h" #include "tensorflow/contrib/lite/model.h" +#include "tensorflow/contrib/lite/tools/gen_op_registration.h" namespace tflite { diff --git a/tensorflow/contrib/lite/tools/gen_op_registration.h b/tensorflow/contrib/lite/tools/gen_op_registration.h index 363bb2335c..318859e23d 100644 --- a/tensorflow/contrib/lite/tools/gen_op_registration.h +++ b/tensorflow/contrib/lite/tools/gen_op_registration.h @@ -16,6 +16,7 @@ limitations under the License. #define THIRD_PARTY_TENSORFLOW_CONTRIB_LITE_TOOLS_GEN_OP_REGISTRATION_H_ #include "tensorflow/contrib/lite/model.h" +#include "tensorflow/contrib/lite/string.h" namespace tflite { |