[CI] Replaced tensorflow with tensorflow-cpu (#3261)

* Replaced tensorflow with tensorflow-cpu

* Use tensorflow-cpu for bazel targets

Author: veblush

WORKSPACE

@@ -101,7 +101,7 @@ py_pkg_cc_deps(
     name = "tensorflow_cc_deps",
     includes = ["tensorflow/include"],
     libs = ["tensorflow/libtensorflow_framework.so.2"],
-    pkg = requirement("tensorflow"),
+    pkg = requirement("tensorflow-cpu"),
 )
 
 # Optimized kernel deps

python/tflite_micro/BUILD

@@ -94,7 +94,7 @@ py_test(
     deps = [
         ":runtime",
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
         "//tensorflow/lite/micro/examples/recipes:add_four_numbers",
         "//tensorflow/lite/micro/testing:generate_test_models_lib",
     ],
@@ -116,7 +116,7 @@ py_test(
     deps = [
         ":runtime",
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
         "//tensorflow/lite/micro/compression",
     ],
 )

python/tflite_micro/signal/BUILD

@@ -74,7 +74,7 @@ py_test(
     deps = [
         ":delay_op",
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 
@@ -105,7 +105,7 @@ py_test(
     deps = [
         ":energy_op",
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 
@@ -136,7 +136,7 @@ py_test(
     deps = [
         ":fft_ops",
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 
@@ -173,7 +173,7 @@ py_test(
     deps = [
         ":filter_bank_ops",
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 
@@ -204,7 +204,7 @@ py_test(
     deps = [
         ":framer_op",
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 
@@ -233,7 +233,7 @@ py_test(
         ":overlap_add_op",
         requirement("absl_py"),
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 
@@ -264,7 +264,7 @@ py_test(
     deps = [
         ":pcan_op",
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 
@@ -295,7 +295,7 @@ py_test(
     deps = [
         ":stacker_op",
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 
@@ -325,6 +325,6 @@ py_test(
     deps = [
         ":window_op",
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )

python/tflite_micro/signal/utils/BUILD

@@ -43,7 +43,7 @@ py_library(
     srcs = ["util.py"],
     visibility = ["//visibility:public"],
     deps = [
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
         "//python/tflite_micro:runtime",
     ],
 )

tensorflow/lite/micro/compression/BUILD

@@ -104,7 +104,7 @@ py_test(
     deps = [
         "metadata_py",
         "@flatbuffers//:runtime_py",
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 
@@ -155,7 +155,7 @@ py_test(
         "//tensorflow/lite/python:schema_py",
         requirement("bitarray"),
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 
@@ -175,7 +175,7 @@ py_test(
     deps = [
         ":model_facade",
         ":test_models",
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 
@@ -193,7 +193,7 @@ py_test(
     srcs = ["spec_test.py"],
     deps = [
         ":spec",
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 
@@ -212,7 +212,7 @@ py_test(
     deps = [
         ":spec",
         ":spec_builder",
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 
@@ -233,7 +233,7 @@ py_test(
     deps = [
         ":test_models",
         "//tensorflow/lite/python:schema_py",
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 

tensorflow/lite/micro/examples/hello_world/BUILD

@@ -50,7 +50,7 @@ py_binary(
     deps = [
         requirement("absl_py"),
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
         "//python/tflite_micro:runtime",
     ],
 )
@@ -72,6 +72,6 @@ py_binary(
     srcs = ["train.py"],
     deps = [
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )

tensorflow/lite/micro/examples/hello_world/quantization/BUILD

@@ -8,7 +8,7 @@ py_binary(
     deps = [
         requirement("absl_py"),
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
         "//python/tflite_micro:runtime",
     ],
 )

tensorflow/lite/micro/examples/micro_speech/BUILD

@@ -210,7 +210,7 @@ py_binary(
     deps = [
         requirement("absl_py"),
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
         "//python/tflite_micro:runtime",
         "//python/tflite_micro/signal:ops",
         "//python/tflite_micro/signal/utils:util",

tensorflow/lite/micro/examples/mnist_lstm/BUILD

@@ -6,7 +6,7 @@ py_binary(
     srcs = ["train.py"],
     deps = [
         requirement("numpy"),
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
     ],
 )
 

tensorflow/lite/micro/examples/person_detection/utils/BUILD

@@ -33,7 +33,7 @@ py_test(
     ],
     deps = [
         ":raw_to_bitmap_lib",
-        requirement("tensorflow"),
+        requirement("tensorflow-cpu"),
         requirement("numpy"),
     ],
 )