Compute argmax only once
Compute argmax only once

OlafLee/LV_groundhog,sebastien-j/LV_groundhog,sebastien-j/LV_groundhog,OlafLee/LV_groundhog,sebastien-j/LV_groundhog,OlafLee/LV_groundhog,sebastien-j/LV_groundhog,OlafLee/LV_groundhog,OlafLee/LV_groundhog,OlafLee/LV_groundhog,sebastien-j/LV_groundhog,sebastien-j/LV_groundhog