Add parameters for 54x54x54 kernel