Tensorflow-cpu package does not support aarch64

[mansnils]

This PR #3261 replace tensorflow with tensorflow-cpu to save memory. Problem is that tensorflow-cpu package is not supported on aarch64.

The tensorflow-cpu package missing aarch64 support issue has been raised before but has unfortunately not been resolved: tensorflow/tensorflow#62673

It has previously been resolved in tflite-micro: #2367

This gives problems on our internal CI running on aarch64.
For example the test_generate_micro_mutable_op_resolver_tests.sh script now depends on tensorflow-cpu.

[veblush]

Thanks for flagging this. I switched to tensorflow-cpu to reduce package size and resolve stability issues in our CI, not realizing it lacked aarch64 support. Since reverting it would break the CI again, I can't roll it back immediately.

While I work on a solution, could you clarify: do your specific build targets depending on tensorflow, or is this just blocking you because it's listed as a dependency?

[mansnils]

It can be reproduced by calling test_generate_micro_mutable_op_resolver_tests.sh without having tensorflow-cpu (or maybe bazel will auto install?). It is essentially bazel calling a python script. If calling the python script directly it works as long as tensorflow is installed.
The script has some dependency (perhaps only this?):
tensorflow/lite/micro/tools/generate_test_for_model.py:import tensorflow as tf
If that can be replaced with tflite_runtime pip package or similar the equivalent bazel depency to tensorflow-cpu can also be replaced, which should resolve this whole issue.