Enhancing neural processing units with digital in-memory computing