You can set the header with product search by navigating through
Dashboard >> Appearance >> Customize >> Desktop header >> Layout.
In the layout section, select the fourth layout.
Choose the option Product Search from the dropdown Product Search Option.