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

Ruy testing.

Browse Source 
PiperOrigin-RevId: 247722924
This commit is contained in:
Benoit Jacob 2019-05-10 21:44:20 -07:00 committed by TensorFlower Gardener
parent 645ce7583c
commit fea8de6c36
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/lite/experimental/ruy/BUILD
View File

@ -380,6 +380,7 @@ ruy_test(
("u8", "u8", "i32", "i16"), ("u8", "u8", "i32", "i16"),
("i8", "i8", "i32", "i32"), ("i8", "i8", "i32", "i32"),
], ],
tags = ["slow"],
) )
ruy_test( ruy_test(

3
tensorflow/lite/experimental/ruy/ruy_test.bzl
View File

@ -6,7 +6,7 @@ corresponding to tuples of types for LHS, RHS, accumulator
and destination. and destination.
""" """
def ruy_test(name, srcs, lhs_rhs_accum_dst): def ruy_test(name, srcs, lhs_rhs_accum_dst, tags = []):
for (lhs, rhs, accum, dst) in lhs_rhs_accum_dst: for (lhs, rhs, accum, dst) in lhs_rhs_accum_dst:
native.cc_test( native.cc_test(
name = "%s_%s_%s_%s_%s" % (name, lhs, rhs, accum, dst), name = "%s_%s_%s_%s_%s" % (name, lhs, rhs, accum, dst),
@ -21,6 +21,7 @@ def ruy_test(name, srcs, lhs_rhs_accum_dst):
"//tensorflow/lite/experimental/ruy:test_lib", "//tensorflow/lite/experimental/ruy:test_lib",
"@com_google_googletest//:gtest_main", "@com_google_googletest//:gtest_main",
], ],
tags = tags,
) )
def ruy_benchmark(name, srcs, lhs_rhs_accum_dst): def ruy_benchmark(name, srcs, lhs_rhs_accum_dst):