diff options
author | Zach Koopmans <zkoopmans@google.com> | 2020-07-28 12:51:54 -0700 |
---|---|---|
committer | gVisor bot <gvisor-bot@google.com> | 2020-07-28 12:56:06 -0700 |
commit | 5873b0f43f5ff4d811f951baaacb7bbe8b1a486c (patch) | |
tree | 564e3300700900c47bd453ebc303e45c8ad19a4e /test/benchmarks/ml/BUILD | |
parent | 15d13e3398d519c48e9278889d7047a937f95d9c (diff) |
Port tensorflow benchmark.
PiperOrigin-RevId: 323633737
Diffstat (limited to 'test/benchmarks/ml/BUILD')
-rw-r--r-- | test/benchmarks/ml/BUILD | 21 |
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", + ], +) |