920bf272c8
RELNOTES=TFLite Hexagon Delegate out of experimental PiperOrigin-RevId: 315925875 Change-Id: Ic286fc1e203c13d712ea1c904dbd928aa7e4d6af
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
"""Rules for generating unit-tests using hexagon delegates."""
|
|
|
|
load("//tensorflow/lite:special_rules.bzl", "tflite_hexagon_mobile_test")
|
|
|
|
def hexagon_op_tests(
|
|
srcs = [],
|
|
deps = []):
|
|
"""Create both monolithic and individual unit test targets for each test file in 'srcs'.
|
|
|
|
Args:
|
|
srcs: list of test files, separate target will be created for each item in the list.
|
|
deps: Dependencies will be added to all test targets.
|
|
"""
|
|
|
|
for src in srcs:
|
|
parts = src.split(".cc")
|
|
native.cc_test(
|
|
name = "hexagon_" + parts[0],
|
|
srcs = [src],
|
|
deps = deps,
|
|
linkstatic = 1,
|
|
tags = [
|
|
"no_oss",
|
|
"nobuilder",
|
|
"notap",
|
|
],
|
|
)
|
|
|
|
all_ops_test_name = "hexagon_op_tests_all"
|
|
native.cc_test(
|
|
name = all_ops_test_name,
|
|
srcs = srcs,
|
|
deps = deps,
|
|
linkstatic = 1,
|
|
tags = [
|
|
"no_oss",
|
|
"nobuilder",
|
|
"notap",
|
|
],
|
|
)
|
|
tflite_hexagon_mobile_test(all_ops_test_name)
|