├── .gitignore ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── dtensor_app_naive.py ├── dtensor_keras_bert.py ├── gpu ├── README.md ├── gce-cluster │ ├── bootstrap.sh │ ├── launch.py.in │ └── requirements.txt ├── gce-node │ ├── bootstrap.sh │ ├── launch │ └── requirements.txt └── make-cluster-commands.sh └── tpu ├── README.md ├── bootstrap.sh ├── make-cluster-commands.sh └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- 1 | .venv/ -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/README.md -------------------------------------------------------------------------------- /dtensor_app_naive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/dtensor_app_naive.py -------------------------------------------------------------------------------- /dtensor_keras_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/dtensor_keras_bert.py -------------------------------------------------------------------------------- /gpu/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/gpu/README.md -------------------------------------------------------------------------------- /gpu/gce-cluster/bootstrap.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/gpu/gce-cluster/bootstrap.sh -------------------------------------------------------------------------------- /gpu/gce-cluster/launch.py.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/gpu/gce-cluster/launch.py.in -------------------------------------------------------------------------------- /gpu/gce-cluster/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/gpu/gce-cluster/requirements.txt -------------------------------------------------------------------------------- /gpu/gce-node/bootstrap.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/gpu/gce-node/bootstrap.sh -------------------------------------------------------------------------------- /gpu/gce-node/launch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/gpu/gce-node/launch -------------------------------------------------------------------------------- /gpu/gce-node/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/gpu/gce-node/requirements.txt -------------------------------------------------------------------------------- /gpu/make-cluster-commands.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/gpu/make-cluster-commands.sh -------------------------------------------------------------------------------- /tpu/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/tpu/README.md -------------------------------------------------------------------------------- /tpu/bootstrap.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/tpu/bootstrap.sh -------------------------------------------------------------------------------- /tpu/make-cluster-commands.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/tpu/make-cluster-commands.sh -------------------------------------------------------------------------------- /tpu/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tensorflow/dtensor-gcp-examples/HEAD/tpu/requirements.txt --------------------------------------------------------------------------------