Prepare other layers to operate on batches.