Use lax.div in avg_pool so integer inputs stay integer - #104

Use lax.div in avg_pool so integer inputs stay integer#104

qihqi merged 1 commit into

google:maingoogle/torchax:mainfrom

benSepanski:avg-pool-integer-divbenSepanski/torchax:avg-pool-integer-divCopy head branch name to clipboard