Conflict with bazel scripts and installed tflite_micro wheel

[mansnils]

This comment summarize the problem: #2545 (comment)

How to reproduce:

Make sure tflite_micro is not installed. Either of these should run successfully,
./tensorflow/lite/micro/tools/ci_build/test_generate_integration_tests.sh
./tensorflow/lite/micro/tools/ci_build/test_generate_micro_mutable_op_resolver_tests.sh

Now pip install tflite_micro and try to run the ci test scripts. It fails, e.g.

    from tflite_micro.tensorflow.lite.micro.tools import generate_test_for_model
ModuleNotFoundError: No module named 'tflite_micro.tensorflow.lite.micro'

[mansnils]

Summarizing the problem, taking generate_micro_mutable_op_resolver_from_model.py as example.

When running the script as regular python script without bazel, it does not work because visualize is not part of the tflite-micro wheel.
When running the script with bazel, bazel can pick up visualize from the path tflite_micro/tensorflow/lite/tools as long as tflite-micro wheel is not installed. If tflite-micro is installed bazel will only look in that, where again visualize is not part of.

So problem is the script does not work when tflite-micro is installed.

Possible solutions:

• Import visualize from tensorflow instead of tflite-micro. That should make it work for both bazel and as a regular python script.
• Add visualize to tflite-micro. That should make it work for both bazel and as a regular python script.
• Make bazel find visualize even if tflite-micro is installed. That will make it work for bazel, which is good enough as far as I am concerned.