wehub-resource-sync
30 lines
849 B
Python
30 lines
849 B
Python
# 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
|
|
],
|
|
)
|