aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorGravatar Yu-Cheng Ling <ycling@google.com>2018-05-14 15:14:53 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-05-14 15:17:52 -0700
commit2bf2308872a6dfa8d6d0809acf0098f666e00fe8 (patch)
tree48cb085b891f3df1a91715a3b66f72f7469f4c90
parent6a2828183040282fa77080912596afc2799dc40b (diff)
Refactoring: Remove lite/tools:mutable_op_resolver dependency.
PiperOrigin-RevId: 196575387
-rw-r--r--tensorflow/contrib/lite/examples/ios/camera/CameraExampleViewController.mm2
-rw-r--r--tensorflow/contrib/lite/examples/ios/simple/RunModelViewController.mm2
-rw-r--r--tensorflow/contrib/lite/models/smartreply/BUILD2
-rw-r--r--tensorflow/contrib/lite/models/smartreply/predictor.cc2
-rw-r--r--tensorflow/contrib/lite/tools/gen_op_registration_main.cc2
5 files changed, 4 insertions, 6 deletions
diff --git a/tensorflow/contrib/lite/examples/ios/camera/CameraExampleViewController.mm b/tensorflow/contrib/lite/examples/ios/camera/CameraExampleViewController.mm
index d74e275f04..59b575ab6e 100644
--- a/tensorflow/contrib/lite/examples/ios/camera/CameraExampleViewController.mm
+++ b/tensorflow/contrib/lite/examples/ios/camera/CameraExampleViewController.mm
@@ -25,8 +25,8 @@
#include "tensorflow/contrib/lite/kernels/register.h"
#include "tensorflow/contrib/lite/model.h"
+#include "tensorflow/contrib/lite/op_resolver.h"
#include "tensorflow/contrib/lite/string_util.h"
-#include "tensorflow/contrib/lite/tools/mutable_op_resolver.h"
#define LOG(x) std::cerr
diff --git a/tensorflow/contrib/lite/examples/ios/simple/RunModelViewController.mm b/tensorflow/contrib/lite/examples/ios/simple/RunModelViewController.mm
index 0ab7aa25d0..32da7f7e4f 100644
--- a/tensorflow/contrib/lite/examples/ios/simple/RunModelViewController.mm
+++ b/tensorflow/contrib/lite/examples/ios/simple/RunModelViewController.mm
@@ -24,8 +24,8 @@
#include "tensorflow/contrib/lite/kernels/register.h"
#include "tensorflow/contrib/lite/model.h"
+#include "tensorflow/contrib/lite/op_resolver.h"
#include "tensorflow/contrib/lite/string_util.h"
-#include "tensorflow/contrib/lite/tools/mutable_op_resolver.h"
#include "ios_image_load.h"
diff --git a/tensorflow/contrib/lite/models/smartreply/BUILD b/tensorflow/contrib/lite/models/smartreply/BUILD
index a82d1f2eb6..8b5fa240ac 100644
--- a/tensorflow/contrib/lite/models/smartreply/BUILD
+++ b/tensorflow/contrib/lite/models/smartreply/BUILD
@@ -22,7 +22,6 @@ cc_library(
"//tensorflow/contrib/lite:framework",
"//tensorflow/contrib/lite:string_util",
"//tensorflow/contrib/lite/kernels:builtin_ops",
- "//tensorflow/contrib/lite/tools:mutable_op_resolver",
"@com_google_absl//absl/strings",
"@com_googlesource_code_re2//:re2",
"@farmhash_archive//:farmhash",
@@ -39,7 +38,6 @@ cc_library(
"//tensorflow/contrib/lite:framework",
"//tensorflow/contrib/lite:string_util",
"//tensorflow/contrib/lite/kernels:builtin_ops",
- "//tensorflow/contrib/lite/tools:mutable_op_resolver",
"@com_google_absl//absl/strings",
"@com_googlesource_code_re2//:re2",
],
diff --git a/tensorflow/contrib/lite/models/smartreply/predictor.cc b/tensorflow/contrib/lite/models/smartreply/predictor.cc
index 6da5cc8eec..ceef8e6a29 100644
--- a/tensorflow/contrib/lite/models/smartreply/predictor.cc
+++ b/tensorflow/contrib/lite/models/smartreply/predictor.cc
@@ -20,8 +20,8 @@ limitations under the License.
#include "tensorflow/contrib/lite/interpreter.h"
#include "tensorflow/contrib/lite/kernels/register.h"
#include "tensorflow/contrib/lite/model.h"
+#include "tensorflow/contrib/lite/op_resolver.h"
#include "tensorflow/contrib/lite/string_util.h"
-#include "tensorflow/contrib/lite/tools/mutable_op_resolver.h"
void RegisterSelectedOps(::tflite::MutableOpResolver* resolver);
diff --git a/tensorflow/contrib/lite/tools/gen_op_registration_main.cc b/tensorflow/contrib/lite/tools/gen_op_registration_main.cc
index 17b514c916..f7df80821f 100644
--- a/tensorflow/contrib/lite/tools/gen_op_registration_main.cc
+++ b/tensorflow/contrib/lite/tools/gen_op_registration_main.cc
@@ -55,7 +55,7 @@ void GenerateFileContent(const std::string& tflite_path,
std::ofstream fout(filename);
fout << "#include \"" << tflite_path << "/model.h\"\n";
- fout << "#include \"" << tflite_path << "/tools/mutable_op_resolver.h\"\n";
+ fout << "#include \"" << tflite_path << "/op_resolver.h\"\n";
fout << "namespace tflite {\n";
fout << "namespace ops {\n";