Add missing optimizations for hltekdi

Change-Id: I05eead04df404ea37c7a4cd46dda395885651bab