blob: bdc35157e937e51393fafa12df792c1f9ab4ce75 [file] [log] [blame]
Vijay Vasudevan8cc567b2016-05-26 11:05:13 -08001workspace(name = "org_tensorflow")
2
Brian Zhao8782b762020-01-13 19:32:03 -08003load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
Brian Zhao9d6198d2020-01-15 11:42:24 -08004load("//third_party:repo.bzl", "tf_http_archive")
A. Unique TensorFlower5b145772018-11-30 11:11:23 -08005
Brian Zhao9d6198d2020-01-15 11:42:24 -08006tf_http_archive(
Justine Tunney02f276c2017-02-03 17:13:49 -08007 name = "io_bazel_rules_closure",
A. Unique TensorFlower265042c2019-06-13 11:02:06 -07008 sha256 = "5b00383d08dd71f28503736db0500b6fb4dda47489ff5fc6bed42557c07c6ba9",
9 strip_prefix = "rules_closure-308b05b2419edb5c8ee0471b67a40403df940149",
Brian Zhao9d6198d2020-01-15 11:42:24 -080010 patch_file = "@org_tensorflow//third_party:rules_closure.patch",
Justine Tunney02f276c2017-02-03 17:13:49 -080011 urls = [
Frank Chen2b5ece22019-07-23 22:21:29 -070012 "https://storage.googleapis.com/mirror.tensorflow.org/github.com/bazelbuild/rules_closure/archive/308b05b2419edb5c8ee0471b67a40403df940149.tar.gz",
A. Unique TensorFlower265042c2019-06-13 11:02:06 -070013 "https://github.com/bazelbuild/rules_closure/archive/308b05b2419edb5c8ee0471b67a40403df940149.tar.gz", # 2019-06-13
Justine Tunney02f276c2017-02-03 17:13:49 -080014 ],
15)
Gunhan Gulsoyb4ae8522018-09-26 02:31:31 -070016
A. Unique TensorFlowerdad41972019-06-24 07:21:59 -070017# Load tf_repositories() before loading dependencies for other repository so
18# that dependencies like com_google_protobuf won't be overridden.
19load("//tensorflow:workspace.bzl", "tf_repositories")
A. Unique TensorFlower508f76b2019-06-19 04:14:12 -070020# Please add all new TensorFlow dependencies in workspace.bzl.
A. Unique TensorFlowerdad41972019-06-24 07:21:59 -070021tf_repositories()
A. Unique TensorFlower508f76b2019-06-19 04:14:12 -070022
A. Unique TensorFlower98ecdee2019-10-05 01:15:18 -070023register_toolchains("@local_config_python//:py_toolchain")
24
Justine Tunney02f276c2017-02-03 17:13:49 -080025load("@io_bazel_rules_closure//closure:defs.bzl", "closure_repositories")
Gunhan Gulsoyb4ae8522018-09-26 02:31:31 -070026
Justine Tunney02f276c2017-02-03 17:13:49 -080027closure_repositories()
28
A. Unique TensorFlower62bc6402018-12-13 00:11:34 -080029load("//third_party/toolchains/preconfig/generate:archives.bzl",
30 "bazel_toolchains_archive")
31
32bazel_toolchains_archive()
33
34load(
35 "@bazel_toolchains//repositories:repositories.bzl",
36 bazel_toolchains_repositories = "repositories",
A. Unique TensorFlower54cc77e2018-11-12 09:31:53 -080037)
38
A. Unique TensorFlower62bc6402018-12-13 00:11:34 -080039bazel_toolchains_repositories()
40
41load(
A. Unique TensorFlowerb0022ad2019-02-06 15:41:09 -080042 "@io_bazel_rules_docker//repositories:repositories.bzl",
A. Unique TensorFlower62bc6402018-12-13 00:11:34 -080043 container_repositories = "repositories",
A. Unique TensorFlower54cc77e2018-11-12 09:31:53 -080044)
45
A. Unique TensorFlower62bc6402018-12-13 00:11:34 -080046container_repositories()
A. Unique TensorFlower54cc77e2018-11-12 09:31:53 -080047
A. Unique TensorFlower62bc6402018-12-13 00:11:34 -080048load("//third_party/toolchains/preconfig/generate:workspace.bzl",
49 "remote_config_workspace")
A. Unique TensorFlower54cc77e2018-11-12 09:31:53 -080050
51remote_config_workspace()
52
A. Unique TensorFlower1942fea2019-04-11 11:23:15 -070053# Use `swift_rules_dependencies` to fetch the toolchains. With the
54# `git_repository` rules above, the following call will skip redefining them.
A. Unique TensorFlower146c2da2018-12-17 15:59:15 -080055load("@build_bazel_rules_swift//swift:repositories.bzl", "swift_rules_dependencies")
56swift_rules_dependencies()
57
Ilya Biryukov9e651e42018-03-22 05:33:42 -070058# We must check the bazel version before trying to parse any other BUILD
59# files, in case the parsing of those build files depends on the bazel
60# version we require here.
61load("//tensorflow:version_check.bzl", "check_bazel_version_at_least")
Gunhan Gulsoyaaea5412019-11-29 22:08:00 -080062check_bazel_version_at_least("1.0.0")
Ilya Biryukov9e651e42018-03-22 05:33:42 -070063
Michael Case51053502018-06-05 17:47:19 -070064load("//third_party/android:android_configure.bzl", "android_configure")
65android_configure(name="local_config_android")
66load("@local_config_android//:android.bzl", "android_workspace")
67android_workspace()
Manjunath Kudlurf41959c2015-11-06 16:27:58 -080068
A. Unique TensorFlower508f76b2019-06-19 04:14:12 -070069# If a target is bound twice, the later one wins, so we have to do tf bindings
70# at the end of the WORKSPACE file.
71load("//tensorflow:workspace.bzl", "tf_bind")
72tf_bind()
Manjunath Kudlurf41959c2015-11-06 16:27:58 -080073
A. Unique TensorFlower5b145772018-11-30 11:11:23 -080074http_archive(
Michael Cased90054e2018-02-07 14:36:00 -080075 name = "inception_v1",
A. Unique TensorFlower5b145772018-11-30 11:11:23 -080076 build_file = "//:models.BUILD",
Michael Cased90054e2018-02-07 14:36:00 -080077 sha256 = "7efe12a8363f09bc24d7b7a450304a15655a57a7751929b2c1593a71183bb105",
Justine Tunney52b4d472017-05-17 16:13:33 -070078 urls = [
Frank Chen57e0d1a2019-07-24 18:43:43 -070079 "https://storage.googleapis.com/download.tensorflow.org/models/inception_v1.zip",
Justine Tunney52b4d472017-05-17 16:13:33 -070080 ],
Dan Smilkovf0e9bd32016-11-09 12:53:00 -080081)
82
A. Unique TensorFlower5b145772018-11-30 11:11:23 -080083http_archive(
A. Unique TensorFlower53aabd52017-08-09 18:18:26 -070084 name = "mobile_ssd",
A. Unique TensorFlower5b145772018-11-30 11:11:23 -080085 build_file = "//:models.BUILD",
A. Unique TensorFlower53aabd52017-08-09 18:18:26 -070086 sha256 = "bddd81ea5c80a97adfac1c9f770e6f55cbafd7cce4d3bbe15fbeb041e6b8f3e8",
87 urls = [
Frank Chen57e0d1a2019-07-24 18:43:43 -070088 "https://storage.googleapis.com/download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_android_export.zip",
A. Unique TensorFlower53aabd52017-08-09 18:18:26 -070089 ],
90)
91
A. Unique TensorFlower5b145772018-11-30 11:11:23 -080092http_archive(
Justine Tunney52b4d472017-05-17 16:13:33 -070093 name = "mobile_multibox",
A. Unique TensorFlower5b145772018-11-30 11:11:23 -080094 build_file = "//:models.BUILD",
Justine Tunney52b4d472017-05-17 16:13:33 -070095 sha256 = "859edcddf84dddb974c36c36cfc1f74555148e9c9213dedacf1d6b613ad52b96",
96 urls = [
Frank Chen57e0d1a2019-07-24 18:43:43 -070097 "https://storage.googleapis.com/download.tensorflow.org/models/mobile_multibox_v1a.zip",
Justine Tunney52b4d472017-05-17 16:13:33 -070098 ],
Andrew Harpa30b9922016-11-28 15:41:52 -080099)
100
A. Unique TensorFlower5b145772018-11-30 11:11:23 -0800101http_archive(
Justine Tunney52b4d472017-05-17 16:13:33 -0700102 name = "stylize",
A. Unique TensorFlower5b145772018-11-30 11:11:23 -0800103 build_file = "//:models.BUILD",
Justine Tunney52b4d472017-05-17 16:13:33 -0700104 sha256 = "3d374a730aef330424a356a8d4f04d8a54277c425e274ecb7d9c83aa912c6bfa",
105 urls = [
Frank Chen57e0d1a2019-07-24 18:43:43 -0700106 "https://storage.googleapis.com/download.tensorflow.org/models/stylize_v1.zip",
Justine Tunney52b4d472017-05-17 16:13:33 -0700107 ],
Dan Smilkov38a4afc2016-08-31 11:56:58 -0800108)
Pete Warden7d01f892017-08-18 09:32:30 -0700109
A. Unique TensorFlower5b145772018-11-30 11:11:23 -0800110http_archive(
Pete Warden7d01f892017-08-18 09:32:30 -0700111 name = "speech_commands",
A. Unique TensorFlower5b145772018-11-30 11:11:23 -0800112 build_file = "//:models.BUILD",
Pete Warden7d01f892017-08-18 09:32:30 -0700113 sha256 = "c3ec4fea3158eb111f1d932336351edfe8bd515bb6e87aad4f25dbad0a600d0c",
114 urls = [
Frank Chen57e0d1a2019-07-24 18:43:43 -0700115 "https://storage.googleapis.com/download.tensorflow.org/models/speech_commands_v0.01.zip",
Pete Warden7d01f892017-08-18 09:32:30 -0700116 ],
117)