If you know it will shut down reliably:
Attach auxiliary fuel source to high pressure pump, bypassing every component of fuel supply system. Test drive with the same conditions that make it shut off. If it fixes it, you have a supply problem. If problem still exists, the problem is in the high...