keras/src/distribution/distribution_lib.py/DataParallel/get_tensor_layout
class DataParallel:    def get_tensor_layout(self, path):
        # For data parallel training, the intermediate state is not changed.
        return None