-
Notifications
You must be signed in to change notification settings - Fork 74k
/
BUILD
29 lines (26 loc) · 849 Bytes
/
BUILD
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# DTensor is a TensorFlow extension that enables large-scale modeling with
# minimal changes to user code.
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
licenses = ["notice"],
)
package_group(
name = "dtensor-users",
packages = [
"//tensorflow_models/google/...",
"//third_party/aimee/clara2_labs/...",
"//third_party/py/jax_tpu_embedding/...",
"//third_party/py/tf_keras/dtensor/...",
],
)
package_group(
name = "dtensor-internal",
packages = [
"//learning/brain/experimental/jax_tpu_embedding/...",
"//learning/brain/public/...",
"//tensorflow/core/...",
"//tensorflow/dtensor/...",
"//tensorflow/python/...",
"//tensorflow_federated/...", # Needs access to C++ API for dtensor device
],
)