summaryrefslogtreecommitdiffhomepage
path: root/test/benchmarks/ml/BUILD
diff options
context:
space:
mode:
authorZach Koopmans <zkoopmans@google.com>2020-07-28 12:51:54 -0700
committergVisor bot <gvisor-bot@google.com>2020-07-28 12:56:06 -0700
commit5873b0f43f5ff4d811f951baaacb7bbe8b1a486c (patch)
tree564e3300700900c47bd453ebc303e45c8ad19a4e /test/benchmarks/ml/BUILD
parent15d13e3398d519c48e9278889d7047a937f95d9c (diff)
Port tensorflow benchmark.
PiperOrigin-RevId: 323633737
Diffstat (limited to 'test/benchmarks/ml/BUILD')
-rw-r--r--test/benchmarks/ml/BUILD21
1 files changed, 21 insertions, 0 deletions
diff --git a/test/benchmarks/ml/BUILD b/test/benchmarks/ml/BUILD
new file mode 100644
index 000000000..2430b60a7
--- /dev/null
+++ b/test/benchmarks/ml/BUILD
@@ -0,0 +1,21 @@
+load("//tools:defs.bzl", "go_library", "go_test")
+
+package(licenses = ["notice"])
+
+go_library(
+ name = "ml",
+ testonly = 1,
+ srcs = ["ml.go"],
+ deps = ["//test/benchmarks/harness"],
+)
+
+go_test(
+ name = "ml_test",
+ size = "large",
+ srcs = ["tensorflow_test.go"],
+ library = ":ml",
+ deps = [
+ "//pkg/test/dockerutil",
+ "//test/benchmarks/harness",
+ ],
+)