Revert "gen_micro_mutable script: update import path" (#2556) Reverts tensorflow/tflite-micro#2545 BUG=broke some internal scripts.
diff --git a/tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver/generate_micro_mutable_op_resolver_from_model.py b/tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver/generate_micro_mutable_op_resolver_from_model.py index 5c91f31..b7fbd9a 100644 --- a/tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver/generate_micro_mutable_op_resolver_from_model.py +++ b/tensorflow/lite/micro/tools/gen_micro_mutable_op_resolver/generate_micro_mutable_op_resolver_from_model.py
@@ -1,4 +1,4 @@ -# Copyright 2024 The TensorFlow Authors. All Rights Reserved. +# Copyright 2023 The TensorFlow Authors. All Rights Reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -23,7 +23,7 @@ from absl import flags from mako import template -from tensorflow.lite.tools import visualize as visualize +from tflite_micro.tensorflow.lite.tools import visualize as visualize TEMPLATE_DIR = os.path.join(os.path.dirname(__file__), 'templates') TEMPLATE_DIR = os.path.abspath(TEMPLATE_DIR)