Use a custom batch size threshold for using a batch-specific kernel.

Use a custom batch size threshold for using a batch-specific kernel. #2