TensorFlow DTensor: Unified API for Distributed Deep Network Training