2020-12-08 00:17:14 +00:00
|
|
|
load("//tools:defs.bzl", "go_library")
|
|
|
|
load("//test/benchmarks:defs.bzl", "benchmark_test")
|
2020-07-28 19:51:54 +00:00
|
|
|
|
|
|
|
package(licenses = ["notice"])
|
|
|
|
|
|
|
|
go_library(
|
|
|
|
name = "ml",
|
|
|
|
testonly = 1,
|
|
|
|
srcs = ["ml.go"],
|
|
|
|
)
|
|
|
|
|
2020-12-08 00:17:14 +00:00
|
|
|
benchmark_test(
|
2020-12-09 19:55:06 +00:00
|
|
|
name = "tensorflow_test",
|
|
|
|
size = "enormous",
|
2020-07-28 19:51:54 +00:00
|
|
|
srcs = ["tensorflow_test.go"],
|
|
|
|
library = ":ml",
|
2020-08-10 21:51:07 +00:00
|
|
|
visibility = ["//:sandbox"],
|
2020-07-28 19:51:54 +00:00
|
|
|
deps = [
|
|
|
|
"//pkg/test/dockerutil",
|
|
|
|
"//test/benchmarks/harness",
|
2021-02-12 20:53:51 +00:00
|
|
|
"//test/benchmarks/tools",
|
2020-07-28 19:51:54 +00:00
|
|
|
],
|
|
|
|
)
|