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

This reverts commit 9cbf6e0403.
This commit is contained in:
Esun Kim 2026-01-12 11:19:37 -08:00 committed by GitHub
parent 6d41f60566
commit fd33340e06
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
21 changed files with 85 additions and 75 deletions

View file

@ -94,7 +94,7 @@ py_test(
deps = [
":runtime",
requirement("numpy"),
requirement("tensorflow-cpu"),
requirement("tensorflow"),
"//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-cpu"),
requirement("tensorflow"),
"//tensorflow/lite/micro/compression",
],
)

View file

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

View file

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