1# DTensor is a TensorFlow extension that enables large-scale modeling with 2# minimal changes to user code. 3 4package( 5 licenses = ["notice"], 6) 7 8package_group( 9 name = "dtensor-users", 10 packages = [ 11 "//tensorflow_models/google/...", 12 "//third_party/aimee/clara2_labs/...", 13 "//third_party/py/keras/dtensor/...", 14 ], 15) 16 17package_group( 18 name = "dtensor-internal", 19 packages = [ 20 "//learning/brain/experimental/jax_tpu_embedding/...", 21 "//learning/brain/public/...", 22 "//tensorflow/core/...", 23 "//tensorflow/dtensor/...", 24 "//tensorflow/python/...", 25 ], 26) 27