experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #46966 from rsun-bdti:cast_zeros_like_micro_build

Browse Source 
PiperOrigin-RevId: 356520941
Change-Id: Id5c7377df1f1fd24ed55eb4f87c2e6ed5e1626f3
This commit is contained in:
TensorFlower Gardener 2021-02-09 09:44:02 -08:00
commit e4f72eb72f
1 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
tensorflow/lite/micro/kernels/BUILD
View File

@ -222,6 +222,19 @@ test_suite(
name = "all_tests",
)
tflite_micro_cc_test(
name = "cast_test",
srcs = ["cast_test.cc"],
deps = [
":kernel_runner",
"//tensorflow/lite/c:common",
"//tensorflow/lite/micro:debug_log",
"//tensorflow/lite/micro:op_resolvers",
"//tensorflow/lite/micro:test_helpers",
"//tensorflow/lite/micro/testing:micro_test",
],
)
tflite_micro_cc_test(
name = "elementwise_test",
srcs = ["elementwise_test.cc"],
@ -248,6 +261,19 @@ tflite_micro_cc_test(
],
)
tflite_micro_cc_test(
name = "zeros_like_test",
srcs = ["zeros_like_test.cc"],
deps = [
":kernel_runner",
"//tensorflow/lite/c:common",
"//tensorflow/lite/micro:debug_log",
"//tensorflow/lite/micro:op_resolvers",
"//tensorflow/lite/micro:test_helpers",
"//tensorflow/lite/micro/testing:micro_test",
],
)
tflite_micro_cc_test(
name = "pooling_test",
srcs = [