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

Add the conv kernel to the BUILD file.

Browse Source
This commit is contained in:
Jens Elofsson 2019-04-03 16:49:01 +02:00
parent c0ba417406
commit 812731d38d
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tensorflow/lite/experimental/micro/kernels/BUILD
View File

@ -13,6 +13,7 @@ load(
cc_library(
name = "micro_ops",
srcs = [
"conv.cc",
"depthwise_conv.cc",
"fully_connected.cc",
"softmax.cc",
@ -133,3 +134,16 @@ tflite_micro_cc_test(
"//tensorflow/lite/experimental/micro/testing:micro_test",
],
)
tflite_micro_cc_test(
name = "conv_test",
srcs = [
"conv_test.cc",
],
deps = [
":all_ops_resolver",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/experimental/micro:micro_framework",
"//tensorflow/lite/experimental/micro/testing:micro_test",
],
)