No it's not based on softmax output. It's single pass for now !