Task-Aware Neural Architecture Search